Windows下Caffe-SSD编译

1、下载Windows版本的Caffe-SSD源码
下载链接:https://github.com/conner99/caffe,选择ssd-microsoft 分支;
2、编译caffe
1)、打开caffe-ssd-microsoft文件夹,然后看到一个windows文件夹,然后继续打开windows文件夹,看到里面一个CommonSettings.props.example文件,复制出来一份,并改名字为CommonSettings.props;
Windows下Caffe-SSD编译_第1张图片
2)、修改CommonSettings.props文件,选择gpu或是cpu以及cuda版本,这里选择的是GPU编译和Cuda7.5;Windows下Caffe-SSD编译_第2张图片
3、编译问题解决
1)、设置libcaffe为启动项,进行编译,报错:与boost相关的regex正则表达式库出现问题,在项目中我们不需要正则表达式,所以将相关语句注释掉即可;
解决方法:
进入项libcaffe中,进入detection_output_layer.hpp中,将与regex相关头文件注释掉;
Windows下Caffe-SSD编译_第3张图片
进入detection_output_layer.cpp文件,将所有出现regex和rv的语句注释掉;
Windows下Caffe-SSD编译_第4张图片
进入detection_output_layer.cu文件,将所有出现regex和rv的语句注释掉;
Windows下Caffe-SSD编译_第5张图片
2)、编译工程,报错:expected an identifier in caffe.pb.h;
解决方法:修改bbox_util.cu,注释掉所有带thrust的语句;
Windows下Caffe-SSD编译_第6张图片
Windows下Caffe-SSD编译_第7张图片
3)、编译工程,报错:找不到”.\caffe\3rdparty\hungarian.h”文件;
编译工程,报错:找不到”.\src\caffe\3rdparty\hungarian.cpp”文件;
解决方法:
在路径.\caffe-master\include\caffe\3rdparty\下添加hungarian.h文件;
在路径.\caffe-master\src\caffe\3rdparty\下添加hungarian.cpp文件;
在这里找这个文件,链接:http://pan.baidu.com/s/1mhYuf7y 密码:3jp2;
4、进行单帧图片目标检测
1)、进入ssd_detect项目添加相关图片显示代码;
Windows下Caffe-SSD编译_第8张图片
2)、编译该项目生成ssd_detect.exe;
Windows下Caffe-SSD编译_第9张图片
3)、新建ssd_detect.bat文件,运行;
ssd_detect.bat文件内容:
(1)ssd_detect.exe路径;
(2)model_file路径,即deploy.prototxt;
(3)weights_file路径,即XXX.caffemodel;
(4)test.txt图片路径,将图片的绝对路径写入test.txt中;
(5)–file_type image,此处还可以进行video检测,可自行看源码修改;
(6)–confidence_threshold 0.5;
Windows下Caffe-SSD编译_第10张图片
运行结果如图所示;
Windows下Caffe-SSD编译_第11张图片

你可能感兴趣的:(caffe)