根据https://github.com/endernewton/tf-faster-rcnn作者写的readme执行,并根据网上查找一些博客,做了整理
安装
1.下载源码:
git clone https://github.com/endernewton/tf-faster-rcnn.git
2.配置环境
tensorflow、cython、pillow、opencv、easydict、matplotlib、scipy、pyyaml
3.查看GPU配置
在lib/setup.py中查看GPU版本,搜索“-arch”查看版本并修改
我使用的是GTX 1080,因此改为’-arch=sm_61’
4.编译Cython
make clean
make
cd ..
5.安装Python COCO API
cd data
git clone https://github.com/pdollar/coco.git
cd coco/PythonAPI
make
cd ../../..
测试预训练模型
1.下载预训练模型
./data/scripts/fetch_faster_rcnn_models.sh