SSD+windows10+GPU/CPU+python+vs2013

注意:编译ssd需要在vs2013上,试了2015每次都出错!!!
参考的博客:这里写链接内容

0 前言
最近在做万恶的目标检测,在试用了fast-dpm之后虽然效果还行,但是速度简直太慢了,于是乎把目标放在了我大深度学习之上,前后使用了faster-rcnn在CPU下大约2s的样子,简直了!看到网上说SSD实时性很好,于是就拿来练手。but,前前后后试了网上很多教程,都是在最后一步测试出错,经过长期的折腾,终于找到原因,希望大家别像我走弯路!!!
SSD原理:

http://blog.csdn.net/u011534057/article/details/52733686

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左右,效果最好)**

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值