一、源码地址:https://github.com/smallcorgi/Faster-RCNN_TF
二、环境搭建
先参考这篇装一下大概环境:https://blog.csdn.net/wwtor/article/details/80603296
然后打开Anaconda Prompt或cmd安装python包(cython、python-opencv、easydict)
先cd到文件目录,然后运行`pip install -r requirements.txt` ,如图
然后cd到/data/coco/PythonAPI ,运行 `python setup.py build_ext --inplace` ,运行`python setup.py build_ext install`
再cd到/lib/utils运行 `python setup.py build_ext --inplace`
三、数据集
1. 将自己的数据转换为VOC格式,看这篇:https://blog.csdn.net/a417197457/article/details/80701944
2. 把自己的xml文件放置在Faster-RCNN-TensorFlow-Python3-master/data/VOCdevkit2007/VOC2007/Annotations路径下,记得把原来的删掉
把jpg文件放在Faster-RCNN-TensorFlow-Python3-master/data/VOCdevkit2007/VOC2007/JPEGImages路径下
把生成的四个txt文档放在Faster-RCNN-TensorFlow-Python3-master/data/VOCdevkit2007/VOC2007/ImageSets/Layout 和 Faster-RCNN-TensorFlow-Python3-master/data/VOCdevkit2007/VOC2007/ImageSets/Main路径下