目标检测SSD+Tensorflow 训练自己的数据集

原文地址:https://blog.youkuaiyun.com/weixin_39881922/article/details/80569803

代码解析:https://blog.youkuaiyun.com/qq1483661204/article/details/79776065

对原文的几点解释这说明:

1.代码地址：https://github.com/balancap/SSD-Tensorflow，下载该代码到本地

注:该代码是github上tensorflow版的SSD star 最多的代码.

2.解压ssd_300_vgg.ckpt.zip 到checkpoint文件夹下

注:ssd_300_vgg.ckpt.zip在checkpoints文件夹下,直接解压到checkpoints即可.

3.测试一下看看，在notebooks中创建demo_test.py,其实就是复制ssd_notebook.ipynb中的代码，该py文件是完成对于单张图片的测试，对Jupyter不熟，就自己改了，感觉这样要方便一些。

注:创建demo_test.py,将下面的73行代码粘贴到里面即可使用.其中,第69行代码可以设置具体使用哪张图像进行测试.

4.将自己的数据集做成VOC2007格式放在该工程下面

注:直接将VOC2007文件夹粘贴到SSD-Tensorflow文件夹下即可.

5. 修改datasets文件夹中pascalvoc_common.py文件，将训练类修改别成自己的

6.  将图像数据转换为tfrecods格式.

7.运行tf_convert_data.py文件