1、下载代码https://github.com/chde222/Faster-RCNN-TensorFlow-Python3
2、安装所依赖包 pip install -r requirements.txt
或者单独利用pip install cython
pip install easydict
3、在 ./data/coco/pythonAPI 下打开cmd运行:
python setup.py build_ext --inplace
python setup.py build_ext install
在./lib/utils下打开cmd 运行:
python setup.py build_ext install
4、下载voc2007数据集。
数据集使用的是VOC2007,下载地址:
http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar
由于被墙,可以下载百度云盘的数据集,链接:https://pan.baidu.com/s/1Y_RzqLvW4CAzTEq4ICFVUA ,提取码:m9dl
在data下新建VOCDevkit2007文件夹。然后
下载文件后,将 VOCtrainval_06-Nov-2007.tar 解压后的文件夹VOC2007放在data/VOCDevkit2007/ 路径下。
5、下载vgg16.ckpt 数据集。
VGG16模型的下载地址:http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz,也可去百度云盘下载,
链接:https://pan.baidu.com/s/11Ty10NJ-rgXkkvM92SVVKw ,提取码:d2jz
下载后将文件重命名为:vgg16.ckpt 然后在data下建立imagenet_weights文件夹,将vgg16.ckpt放入
6、打开pycharm。 运行train.py 文件即可
参考自https://blog.csdn.net/ytusdc/article/details/80255920
https://blog.csdn.net/kellyroslyn/article/details/92159004
7、如果跑demo.py的话,显示下面的错误。
这个地方入坑很久,下载的vgg16模型里就是没有这个文件的,查了好久才知道,这应该是新版TensorFlow模型训练时候产生的文件。作者没有提供,没有办法,只有自己训练。通过上一步的训练能够得到模型。把这些模型都放在这个output\vgg16\voc_2007_trainval+voc_2012_trainval\default路径下。out目录跟demo.py同目录。
demo.py 默认使用的是res101,而我们现在用的是vgg16模型,所以更改demo.py 第108行,deault = vgg16。还有就是模型文件的一些命名可能跟demo.py 文件中的名字不一致,导致找不到文件,自己更改模型文件名,或者demo.py即可。
https://blog.csdn.net/u010554381/article/details/86233339