本文是记录了笔者在win7 vs2013 cuda7.5(无效) cuda6.5 下运行rbg大神的matlab版
ShaoqingRen大神的faster rcnn的一个过程。虽然很多人觉得简单,但是不同的硬件造就了不同的难度,我是在借了超哥的q4000显卡的情况下跑通的。
1 序言
Faster RCNN无比著名的object detect的算法,一直没有去跑下人家的demo,无奈自己的懒癌……这阵子从头来过机器学习的算法,到了这里,没有理由再为自己的懒惰付账了。于是就有了我的一系列调试。在此谢过指导性的博客以及git的博主们以下感谢不分先后:
1 首先是rgb大神 https://github.com/rbgirshick/py-faster-rcnn
2 shaoqingren大神 https://github.com/ShaoqingRen/faster_rcnn
3 http://blog.csdn.net/zhjm07054115/article/details/50405540
4 http://blog.csdn.net/sinat_30071459/article/details/50546891
当然很多东西要去这4个地儿下载哦。
2 下载程序包
你所要下载的包括以下几个内容
1 faster rcnn程序 我是在win7下的matlab2015b调试的因此,你需要去
https://github.com/ShaoqingRen/faster_rcnn
2 caffe_mex 也是shao大神提供的
http://pan.baidu.com/s/1i3m0i0H
看到有人说可以自己编译,我是没找到sln文件,确实不明白在win下边如何编译(明白的给我留言说下,不胜感激),只好老老实实的装了cuda6.5 于是就有了我的一系列心酸的过程。
3 你的output文件夹下的模型和测试文件
http://pan.baidu.com/s/1boOzZtH
3 安装cuda6.5
这里忍不住要吐槽一下,cuda7.5我的gtx960都用的妥妥的,cuda6.5居然不识别我的设备……真真的无语了……热风吹过,我无奈的想淘个2手的970用用……突然想到了超哥,有希望了,他的图形工作站显卡q4000 16g内存……
下载这个版本就o了,只安装toolkit,什么sample啊,3d啊都不需要……装到哪个盘都可以,关键是需要你的环境变量
注意cuda_path就可以了。
4 matlab文件的路径
见下图
需要修改的内容我已经标注,例如我的vs2013装在d盘,cuda_path到不需要改。
5编译及测试
先把第一步下载的内容拷贝到相应的位置,
1 caffe_mex 到external
2 faster-rcnn-final到 output
3 测试图片到faster-rcnn-master根目录
4 运行m文件
依次运行以下
1 faster_rcnn_build.m 编译
2 startup.m 配置路径
3 script_faster_rcnn_demo.m 运行demo,这里注意如果显卡达不到970以上的话,运行
而不是vgg,程序默认的不是zf,修改跟下行一样就行。
这里十分感谢超哥,借我他的专业显卡q4000使用。不然我会苦逼而无所得的。同时也感谢实现了faster rcnn的rbg大神以及shao大神的辛勤工作,让我们领略到了机器学习的魅力。
我测试了下几张自己的照片,也是可以识别的不错