caffe SSD 自己数据集训练+调用模型摄像头时实检测并保存视频

编译和制作自己数据集可参考下面的文章，类似文章很多，此处不再赘述

https://blog.csdn.net/edogawachia/article/details/81669834

https://blog.csdn.net/u013738531/article/details/58637760

这里主要介绍训练注意事项以及利用训练好的caffe模型摄像头实时检测并保存检测视频

训练程序为/examples/ssd/ssd_pascal.py

82行：train_data路径；
84行：test_data路径；
237-246行：model_name、save_dir、snapshot_dir、job_dir、output_result_dir路径；
259-263行：name_size_file、label_map_file路径；
266行：num_classes修改为1 + 类别数;
285行： gpus=”0,1,2,3” ===> 改为”0”
360行：num_test_image：测试集图片数目
如果出现 out of memory，则将batch size 相应改小一些

此时注意要在caffe根目录下运行

python ./examples/ssd/ssd_pascal.py 

这里笔者遇到了一个麻烦的问题，（没问题请忽略这一段）我的cuda安装的时候加了sudo，运行此程序的时候亦需要加sudo
执行sudo python ./examples/ssd/ssd_pascal.py 后发现找不到caffe模块

这是因为sudo python 和python根本就不是一个python，最后在高人的指导下解决了问题，方法如下：

1.任意位置新建一个mypath.pth文件

2.写入/home/wang/caffe/python 根据情况写入自己的路径

3.加权限拷贝到/usr/lib/python2.7/dist-packages中

这时sudo python 就自动选用路径中的python了
（这里搞了好久，千万别乱加权限装库）

执行后就开始训练了

训练完成后，调用模型开始测试。

我标注了30张图片，15张车，15张人，模型用来识别人和车

这是摄像头实时检测后保存的视频截图

代码在这里realtime.py

只需要更改几个路径就可以用了

没有积分的同学可以联系我QQ125070473