1在windows上面训练和测试ssd的方法很少,而且配置也相当之麻烦,搞了几天才调试成功。现在写一下过程分享到博客上,顺便也可以作为今后再次安装的参考。我的配置环境:
windows: 10
gpu: Titan X
cuda: 8.0
caffe: caffe-ssd-microsoft
python: 2.7.13
Caffe-SSD的编译与配置
编译这部分不打算详细展开,网上有很多教程,CPU的较多,下面会给出好用的参考教程。这里推荐一下caffe-ssd-microsoft版本而不是caffe-windows加上caffe-ssd的修改版(网上一些教程是这么改的),两个方法都可以,我这里采用的就是单用caffe-ssd-microsoft的版本。
caffe-ssd-microsoft下载:
配置的参考博客链接:
http://blog.csdn.net/buaalei/article/details/54668507
http://www.jianshu.com/p/9a84cc434e05
这两个可以配合着看,尤其GPU部分推荐看第二个博客。反正我前前后后不停地编译不停地改。
配置caffe-ssd-microsoft
编译libcaffe模块
这个就参考上面给的博客就可以了,我就遇到下面几个问题:
1. 添加3rdparty的hungarian.h和cpp
在这里找这个两个文件,链接:http://pan.baidu.com/s/1mhYuf7y 密码:3jp2
2. 注释掉 detection_output_layer.hpp和detection_output_layer.cu和detection_output_layer.cpp有关regex 的引用和语句
3. C/C++ –> 常规 里的将警告提示为错误修改 为 否
编译caffe模块
libcaffe编译成功后,再进行caffe的编译,注意点就是要添加opencv的引用目录和库目录。这个不难。
编译pycaffe模块
这个可选,如果想用python接口的话这个是必须编译的,注意的是首先要配置好下面两个库的引用和库目录
1. python库目录
2. opencv库目录
之后进行google的protobuf的配置,这个不配置会报错的。
配置protobuf
配置参考:
- 选择python2的环境
下载protobuf.3.0.0:
下载链接:下载两个包:protobuf-python-3.0.0.zip 以及 protoc-3.0.0-win32.zip
protobuf-python-3.0.0为protobuf的安装包
protoc-3.0.0-win32包含protobuf的编译器protoc的win32版本,用以编译*.proto文件。
下载后将protoc-3.0.0-win32\bin\protoc.exe 复制到protobuf-3.0.0\src\
用cmd进入到protoc-3.0.0\python目录,运行 python setup.py build,生成大量*.py,如下命令
python setup.py build
然后再分别写入下面两个命令就可以了。