Windows下Faster R-CNN 配置/Matlab版本编译

配置环境: Windows10x64 Matlab2015Ra VS2013 Opencv2.4.11 CUDA7.5 GTX950M

  1. CUDA7.5安装
    因为Cuda7.5做了很大的优化改进,而且对win10支持较好,所以这里选择安装Cuda7.5,具体过程 参考这里。但是由于作者Matlab 版本的Faster r-CNN的cnn库是在Cuda6.5下编译的,所以这里会出现问题,要么需要自己在Cuda7.5下编译,要么可以在这里下载。

  2. 下载 FasterR-CNN程序包
    在作者ShaoqingRen的github上下载Matlab版本的faster-cnn:链接。然后解压到本地。

  3. 编译external/caffe库
    如果你的电脑安装的是CUDA6.5, 那么可以直接运行fetch_data/fetch_caffe_mex_windows_vs2013_cuda65.m。
    如果你是像本人那样安装了CUDA7.5, 那么你可以在这里下载编译好的库,直接解压到作者代码的根目录下。

  4. 生成nms mex文件
    运行文件根目录下的:faster_rcnn_build.m
    注意:这里运行的时候会经常报错,是需要将functions/nms/nvmex.m下的Cuda_path改成你电脑的CUDA安装路径。

  5. 设置相关函数路径
    运行startup.m文件,这只函数运行的相关路径。

  6. 下载CNN的models
    要么运行作者代码包里的:fetch_data/fetch_faster_rcnn_final_model.m 自动下载并保存在output文件夹里面。
    或者自己在百度云里下载,并解压到output文件夹里。

  7. 测试运行demo
    如果前面一切ok没有问题的话,这里将experiments/script_faster_rcnn_demo.m 拷贝到根目录下,然后运行就可以看到结果了。
    注意:由于VGG16模型太大了,所以一般电脑运行起来会出现matlab奔溃,所以这里要么将
    opts.use_gpu = false;在cpu下运行。
    或者使用ZF模型(比VGG16简单,准确度降低):

    model_dir= fullfile(pwd, 'output', 'faster_rcnn_final', 'faster_rcnn_VOC0712_ZF');

这样就可以看到测试图片的运行结果和运行时间了。
demo运行结果

关于‘invalid Mex File’几点补充

 一般是由于系统找不到库文件的原因引起的。
   1. 检查caffe_faster_rcnn下面的dll库是否完整,不完整就重新拷过来;
   2. 检查cuda, opencv的安装是否完成;
   3. 检查系统环境变量,尤其是在Path中添加cuda, opencv的路径。

by Dr. Sword.

  • 2
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 67
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值