主要参考https://blog.youkuaiyun.com/yexiaogu1104/article/details/77415990
也参考了https://github.com/balancap/SSD-Tensorflow/issues/79
原博主的步骤如下:
1. 下载模型ssd_300_vgg,存放在SSD-Tensorflow-master/checkpoints/这个目录下
2. 打开终端,在主目录 SSD-Tensorflow-master 下,运行cd notebooks/进入notebook目录,运行 jupyter-notebook进入http://localhost:8892/tree服务器,打开 ssd_notebook.ipynb文件
3. 配置一些路径:第4个cell中from notebooks import visualization改成import notebooks,不改的话目录结构不对,会报错
4. 使用自己的图片,改变path = '../demo/'成自己的图片所在路径
5. 从头到尾运行所有cell即可。
说明几点:
1.模型ssd_300_vgg下载需fq,请自备工具
2.from notebook import visualization 可以不改,我没改,也运行成功
3.最重要的一点,在运行cell6的时候,原图是这样的

但是运行的时候会出现

即左上角有一个*号,这是说明未加载成功,这时我们需要将下载好的ssd_300_vgg.zkpt.zip解压,并把其中的两个文件放在ssd_300_vgg目录下,我的是这样

或者把ssd_300_vgg.ckpt.zip解压后里面的两个文件直接放在checkpoints里,也可以解决问题
然后将jupyter notebook中的源代码改成如下形式,再次运行cell时没出现*号就好了

这样就好了,剩下的运行cell就好了,

自己踩的坑,希望对别人有点用。

本文详细介绍如何在SSD-TensorFlow项目中部署预训练的SSD 300 VGG模型进行目标检测,包括模型下载、环境配置、路径设置及常见问题解决等关键步骤。
584

被折叠的 条评论
为什么被折叠?



