Faster-RCNN实现物体检测

代码地址:https://github.com/smallcorgi/Faster-RCNN_TF

软件要求:tensorflow 和python包:cpython、python-opencv、easydict

python包安装命令行:sudo apt-get install cython python-opencv

                                 sudo pip install easydict

硬件要求:For training the end-to-end version of Faster R-CNN with VGG16, 3G of GPU memory is sufficient (using CUDNN)

demo运行步骤  

1、获取代码 

git clone --recursive https://github.com/smallcorgi/Faster-RCNN_TF.git
2、建立cyphon模块 
cd $FRCN_ROOT/lib
make  (在下载的faster rcnn根目录下的lib文件下执行make)
3、 下载在PASCAL VOC 2007上训练的模型去测试 

模型地址:https://drive.google.com/open?id=0ByuDEGFYmWsbZ0EzeUlHcGFIVWM

                 https://www.dropbox.com/s/cfz3blmtmwj6bdh/VGGnet_fast_rcnn_iter_70000.ckpt?dl=0

4、运行demo

cd $FRCN_ROOT (在faster rcnn根目录下运行)
python ./tools/demo.py --model model_path

运行demo过程中遇到的错误

1、g++: error: roi_pooling_op.cu.o: No such file or directory

解决:Setting the following should make it work (it worked at least for me)
(1) export
PATH=$PATH:/usr/local/cuda-8.0/bin/ (or your corresponding cuda path) for your terminal
(2) In make.sh file inside lib directory, do
CXXFLAGS='-D_MWAITXINTRIN_H_INCLUDED'

2、importerror: no module named gpu_nms

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值