windows下faster-rcnn遇到的检测框重复很多问题

    最近在做目标检测。于是就采用了RBG大神的faster rcnn进行切入。从RCNN开始,一直到Faster rcnn,先把理论过了一遍。接下来就是实践了,准备跑下代码。faster rcnn看了大体有两个类型,matlab版本和python版本(好像还有C++版本,不过我没试过)。接下来总结下遇到的一些问题。我的环境是,win7 64位,python2.7,Anconda2,caffe,matlab2016a,VS2013,CUDA7.5,cudnn5.0,显卡GT630(自带的显卡)

    首先是matlab版本的问题。跑matlab遇到找不到我的VS2013编译器,只能找到VS2010。想了很多方法,包括重装matlab等,还是无济于事。后来想到忘了检查环境变量,结果发现正是环境变量的问题,导致找不到编译器。所以遇到这个问题,还是首先检查下环境变量。跑matlab版本renshaoqing的代码,还是比较顺利的,只是matlab经常会出现崩溃的问题(虽然我用的已经是ZFmodel了),所以建议还是用python比较好。matlab版本结果如下图:

    接下来是python版本,我遇到的问题很多。不过主要是最终检测时,出现重复框很多的问题。如下图所示

这个问题初步判断是,nms非极大值抑制出了问题。后来在demo.py中调了阈值,发现毫无反应,大概就是nms没编译好了。接下来就是想着怎样编译好nms这块,看了很多博客,有些人也出现这个问题,但后续没处理。后来看到JiangWu的这篇博客http://www.cnblogs.com/deeplearning1314/p/7137737.html。虽然他是跑cpu版本的,但也有相同之处。结果发现,我比它少了几个步骤。

    (1)我在下载py-faster-rcnn-master时,链接:https://github.com/rbgirshick/py-faster-rcnn,直接下载,没有点进去再去下载caffe-faster-rcnn,导致下载后的caffe-faster-rcnn内部是空的,少了很多东西。于是重新点进去如下图所示,下载了caffe-fast-rcnn。这一步很多人好像没注意。

    (2)第二个地方是lib中的nms模块了,只需下载一个lib文件夹来替换,用lib下的nms文件夹直接替换你的py-faster-rcnn-master\lib 目录下nms文件夹,替换后重新python setup_cuda.py install。编译完成后,将经过上述编译两个py文件后产生的两个后缀为.pyd的文件从py-faster-rcnn-master\lib\build\lib.win-amd64-2.7\nms目录下复制拷到py-faster-rcnn-master\lib\nms中。下载链接:链接:https://pan.baidu.com/s/1_xBiIfWH3fPNNMF2x0TENw 密码:1o8y。

    这是我在py-faster-rcnn中遇到的几个问题。最终结果如下图所示:

    今天在自己的笔记本上跑faster-rcnn,只有cpu。遇到一个新的问题,就是matplotlib 报错,DLL load fail。该问题大概率是matplotlib版本的问题,于是卸掉anacodna自带的matplotlib,在重新下了一个matplotlib进行安装,发现顺利解决问题。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值