注意:编译ssd需要在vs2013上,试了2015每次都出错!!!
参考的博客:这里写链接内容
0 前言
最近在做万恶的目标检测,在试用了fast-dpm之后虽然效果还行,但是速度简直太慢了,于是乎把目标放在了我大深度学习之上,前后使用了faster-rcnn在CPU下大约2s的样子,简直了!看到网上说SSD实时性很好,于是就拿来练手。but,前前后后试了网上很多教程,都是在最后一步测试出错,经过长期的折腾,终于找到原因,希望大家别像我走弯路!!!
SSD原理:
1 所需文件
windows-caffe地址:链接:https://github.com/conner99/caffe
ssd_new替代文件地址:链接:http://pan.baidu.com/s/1mhYuf7y 密码:3jp2
(这个好像自己生成nuget的话就不需要)libboost地址:链接:http://pan.baidu.com/s/1pLbBMpP 密码:41tq
cudnn地址:链接:http://pan.baidu.com/s/1o8vWBhw 密码:kdj4
ssd_detect.cpp(检测视频的,想测试图片把file_type改成image)地址:链接: https://pan.baidu.com/s/1kUAVGhx 密码: zu5g
io.cpp地址:链接:http://pan.baidu.com/s/1i4CL7QP 密码:g623
SSD_300x300模型,测试的时候用,链接: https://pan.baidu.com/s/1kVwsQij 密码: 8adi
ssd源码和很多model的下载地址:https://github.com/weiliu89/caffe/tree/ssd#models 下载好的model记得修改里面的step和offset删掉,且删除最后面的labelmap_xxxx_prototxt文件
2 caffe的配置
一、编译windows-caffe
1、修改commonsetting
注意如果是gpu要对应修改gpu的计算能力参数;
2、编译libcaffe(视警告为错误-否)
二、移植ssd至windows-caffe
1、复制ssd_new下所有文件(除build,data,example以及models之外,好像没有这几个,无所谓啦)至windows-caffe,替换原来windows下对应文件
2、 选中libcaffe,添加文件
在libcaffe工程目录下的include->layers添加现有文件,添加caffe-master->include->caffe->layers下的所有h和hpp文件(除roi_pooling_layer.hpp以外)。
在libcaffe工程目录下的src->layers添加现有文件,添加caffe-master->src->caffe->layers下的所有cpp文件(除roi_pooling_layer.cpp以外)。
在libcaffe工程目录下的include->util添加现有文件,添加caffe-master->include->caffe->util下的所有h和hpp文件。
在libcaffe工程目录下的src->util添加现有文件,添加caffe-master->src->caffe->util下的所有cpp文件。
rebuild only libcaffe,会出现一大堆错误’mkdir’: identifier not found,修改db_lmdb.cpp中
CHECK_EQ(mkdir(source.c_str(), 0744), 0) << "mkdir " << source << " failed";
修改为:
#ifdef MSC_VER
CHECK_EQ(mkdir(source.c_str(), 0744), 0) << "mkdir " << source << " failed";
#endif
继续编译,出现错误error C2360: initialization of ‘occurrences_32’ is skipped by ‘case’ label,在每个case语句添加大括号{}
例如:
case H5T_FLOAT:
LOG_FIRST_N(INFO, 1) << "Datatype class: H5T_FLOAT";
break;
改为:
case H5T_FLOAT:
{
LOG_FIRST_N(INFO, 1) << "Datatype class: H5T_FLOAT";
break;
}
修改common.cpp文件,在文件开头添加:#include
#if defined(_MSC_VER)
#include <io.h>
#endif
在io.cpp第65行中,将:
int fd = open(filename, O_RDONLY);
改成:
int fd = open(filename, O_RDONLY|O_BINARY);
编译libcaffe成功。
c. 设置caffe,ssd_detect项目的properties,在VC++ directories下,添加boost_1_59_0\boost路径至include directories,添加boost_1_59_0\lib64-msvc-12.0路径至library directories(我测了下,好像自己编译nuget时候就不需要添加这个)。
d. 依次编译ssd检测程序。在ssd_detect项目下,下载我修改的ssd_detect.cpp文件,博客上面所需文件有;
e. 测试CPU版本的ssd检测。下载SSD_300x300模型
f.这时候如果运行会出现caffe.priorBoxParameter has no filed named step……等待,注释掉deploy.prototxt文件中所有的step和offset;
g.命令窗口到ssd_detect.exe所在目录,运行
classification.exe -confidence_threshold 0.6 SSD_300x300\deploy.prototxt SSD_300x300\VGG_VOC0712_SSD_300x300_iter_120000.caffemodel SSD_300x300\list.txt
则会输出检测到目标的种类和位置信息,并在图上汇出目标框。
在300*300下大概是30ms,只是出现漏检严重,使用07++12+COCO: SSD300(约90ms,效果略好), SSD512(130ms左右,效果最好)**