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; 
这里写图片描述  
2)、修改CommonSettings.props文件,选择gpu或是cpu以及cuda版本,这里选择的是GPU编译和Cuda7.5; 这里写图片描述  
3、编译问题解决  
1)、设置libcaffe为启动项,进行编译,报错:与boost相关的regex正则表达式库出现问题,在项目中我们不需要正则表达式,所以将相关语句注释掉即可; 
解决方法: 
进入项libcaffe中,进入detection_output_layer.hpp中,将与regex相关头文件注释掉; 
这里写图片描述  
进入detection_output_layer.cpp文件,将所有出现regex和rv的语句注释掉; 
这里写图片描述  
进入detection_output_layer.cu文件,将所有出现regex和rv的语句注释掉; 
这里写图片描述  
2)、编译工程,报错:expected an identifier in caffe.pb.h; 
解决方法:修改bbox_util.cu,注释掉所有带thrust的语句; 
这里写图片描述  
这里写图片描述  
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项目添加相关图片显示代码; 
这里写图片描述  
2)、编译该项目生成ssd_detect.exe; 
这里写图片描述  
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; 
这里写图片描述  
运行结果如图所示; 
这里写图片描述

猜你喜欢

转载自blog.csdn.net/maweifei/article/details/81044783
今日推荐