faster rcnn winodws cpu配置笔记

来源:互联网 发布:拉勾网简历被转发 知乎 编辑:程序博客网 时间:2024/06/06 23:57

参考 http://blog.csdn.net/mr_curry/article/details/54745116

1、下载caffe-master
Matlab版本的Faster RCNN
faster_rcnn_final_model.zip

2、打开CommonSettings.pros 修改
(1)选用cpu版本
这里写图片描述
(2)选用MatlabSupport
这里写图片描述
(3)修改Matlab路径
这里写图片描述

3、打开Caffe.sln工程,在libcaffe文件中添加
roi_pooling_layer.cpp、roi_pooling_layer.cu.
重新生成libcaffe,再生成matcaffe。

在caffe-master\Build\x64\Release文件夹下会有一个matcaffe的文件夹。

然后将
..\caffe-master\Build\x64\Release
中的所有文件复制到
..\faster_rcnn-master\external\caffe\matlab文件夹中。

4、将faster_rcnn_final_model.zip解压至
..\faster_rcnn-master\experiments文件夹下

5、用Matlab打开..\faster_rcnn-master\experiments文件夹中的script_faster_rcnn_demo.m做如下修改

%% -------------------- CONFIG --------------------opts.caffe_version          = 'caffe_faster_rcnn';%注释opts.gpu_id                 = auto_select_gpu;%注释active_caffe_mex(opts.gpu_id, opts.caffe_version);caffe.set_mode_cpu;%添加opts.per_nms_topN           = 6000;opts.nms_overlap_thres      = 0.7;opts.after_nms_topN         = 300;              opts.use_gpu                = false;%%opts.test_scales            = 600;

建议注释掉上面一行(很慢),使用faster_rcnn_VOC0712_ZF 比较快

%% -------------------- INIT_MODEL --------------------%model_dir                   = fullfile(pwd, 'output', 'faster_rcnn_final', 'faster_rcnn_VOC0712_vgg_16layers'); %% VGG-16model_dir                   = fullfile(pwd, 'output', 'faster_rcnn_final', 'faster_rcnn_VOC0712_ZF'); %% ZF

6、回到Commond Window,运行:

addPath(..\faster_rcnn-master\experiments')script_faster_rcnn_demo

效果图:
这里写图片描述
这里写图片描述

原创粉丝点击