faster-rcnn-demo

关注

原创最新推荐文章于 2026-06-17 08:04:23 发布 776 阅读 AI 写同款· GEO 优化 ›

tf-faster-rcnn在cpu下运行demo、测试和训练自己的数据集 1、环境配置官方给出的Faster R-CNN的代码是caffe框架下的，我对caffe不是很了解，一直用的是tensorflow环境，所以去网上找了一下用tensorflow实现的Faster R-CNN。找到了一篇博客https://www.cnblogs.com/toone/p/8433581.html，作者使用的是一个github上Xinlei Chen的tensorflow版本的fa... 阅读详情

原文地址：lefenger.com

运行github上的python版本的faster-rcnn demo的过程为：

克隆github的项目到本地 git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git
在caffe-fast-rcnn文件夹中执行
cp Makefile.config.example Makefile.config
将makefile.config中的

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include

LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib

替换为

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/


LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial/

将makefile.config中的WITH_PYTHON_LAYER :=1前面的注释去掉。
如果你的opencv版本为3.x，将OPENCV_VERSION：=3前面的注释去掉。
在lib文件夹中执行 make。
在caffe-fast-rcnn文件夹中执行 make -j32 && make pycaffe。
在py-faster-rcnn中执行 ./data/scripts/fetch_faster_rcnn_models.sh 下载模型，若下载失败则打开此文件找到链接自行下载。
在py-faster-rcnn中执行 ./tools/demo.py 运行demo。如果出现Check failed：error==cudaSuccess（2 vs.0）out of memory的错误，很遗憾，显存不够，有如下两种办法：（1）执行./tools/demo.py –net zf 改用zf模型测试，因为这个模型层数很少。（2）打开lib/fast_rcnn/config.py , 修改其中的 TRAIN.BATCH_SIZE。

参考： http://blog.youkuaiyun.com/u014380165/article/details/72704604?utm_source=itdadao&utm_medium=referral

收录于