参考:1)目标检测SSD+Tensorflow 训练自己的数据集 - zcc_TPJH - 优快云博客 https://blog.youkuaiyun.com/weixin_39881922/article/details/80569803
2)SSD-Tensorflow训练总结 - 你的朋友不及格,你很难过;你的朋友考了第一,你更难过。 - 优快云博客 https://blog.youkuaiyun.com/w5688414/article/details/78395177
代码链接:GitHub - balancap/SSD-Tensorflow: Single Shot MultiBox Detector in TensorFlow https://github.com/balancap/SSD-Tensorflow
下载下来将先ssd_300_vgg.ckpt.zip解压到当前目录下同时还需要下载vgg_16预训练模型。
准备好数据集VOC2007数据集解压后如图
在运行tf_convert_data.py之前需要传参数,主要是指定相关文件路径,可以直接在py文件中改,也可以在LINUX系统下通过touch xxx.sh命令新建一个xxx.sh文件然后将路径或者其他参数复制进去,再运行sh文件就可以啦。我直接新建了一个tf_convert_data.sh文件,输入以下参数: