参考博客:https://blog.csdn.net/miss_yuki/article/details/79550030
这个博客里面关于ssd-caffe的配置讲述的已经非常详细,我就是按照这个步骤一步一步配置好了所有的内容。不过中间遇见了一些这篇博客里面没有的问题,做一个简单的记录。向大佬看齐~~
- 下载测试训练的数据集:(这里我找的是一个镜像网站下载的)
http://host.robots.ox.ac.uk:8080/pascal/VOC/voc2012/#data
- make runtest -j 16
这个我没有make,因为看博客说是非必须进行的一步。其实是因为一直会出现out of memory的错误,而我不能解决啦~~我查了很多资料都说可以通过该batch_size或者通过nvidia-smi查看GPU使用情况,然后在ssd_pascal.py里面修改gpus的值,但是我都尝试了一直没能解决,也期待看到的大佬不吝赐教~后面我会继续查找问题所在,先挖个坑~~
- 生成LMDB文件
在运行create_list.sh时候没有出现任何问题,但是运行creat_data.sh时出现了ImportError:No module named caffe.proto。根据博客里面介绍的方法,使用vim ~/.bashrc查看隐藏文件,并且把地址修改为caffe-ssd中的python地址。这里我就把地址搞错了,结果因此耽误了很多时间,注意是caffe-ssd里面的python地址,就是你下载的ssd里面的python所在的地址。
总结下来,似乎没有什么值得说的问题,但就是这几个小小的问题一直耽误了我一天的时间,还是自己菜啊~~