faster r-cnn(tensorflow版本)训练VOC数据集、测试、检测指南
使用的faster r-cnn代码github地址
https://github.com/endernewton/tf-faster-rcnn
git下载:
git clone https://github.com/endernewton/tf-faster-rcnn.git
1.准备工作
首先根据自己的显卡调整默认计算能力,打开/lib/setup.py文件,第130行的'-arch=sm_52'改变成你自己显卡的计算能力值
比如本文使用的显卡是gtx1080ti,就改成'-arch=sm_61'
具体计算能力从如下地址查找
https://blog.csdn.net/huixingshao/article/details/82215703
之后进行编译,依次执行
cd tf-faster-rcnn/lib
make clean
make
cd ..
2.训练VOC数据集
将下载好的VOC数据集放在/data文件夹下面
/data/VOCdevkit2007/VOC