【Win10+Tensorflow】Faster-RCNN训练和测试demo

一、系统环境说明

win10+CUDA10.1+cudnn-v7.6+Python3.7
Tesorflow环境配置这里就不写了，默认已经配置好了。

二、下载Faster-RCNN开源代码

网页地址链接：地址
或者在我的百度云盘下载，提取码：7hg9。
这个版本支持python3（3.5/3.6/3.7），Windows和Linux同样支持。

三、Faster-RCNN训练

1、 运行下面代码以安装必须的库（cython,opencv-python,easydict）

pip install -r requirements.txt

2、 切换路径到 ./data/coco/PythonAPI

运行python setup.py build_ext –inplace

运行 python setup.py build_ext install

切换路径到 ./lib/utils

运行 python setup.py build_ext –inplace

3、 下载 pycoco数据集

链接：https://pan.baidu.com/s/1JwWuH6DOwhLAdieJznI3Ag

提取码：dn4p

下载完成后将其解压在 .\data\VOCDevkit2007\VOC2007文件夹下

4、 下载预先训练过的VGG16模型，

链接：https://pan.baidu.com/s/1NZhDIEQRRr425m9-VvJNuQ

提取码：j1id

将其放在 ./data/imagenet_weights路径下，修改名字为“vgg16.ckpt”

5、 运行 python train.py开始训练

训练完成后会生成一个default文件夹，里面保存了生成的模型。

四、测试demo

1、修改demo.py文件

将70000改为40000，因为之迭代了40000次

将红色框内的res101换为vgg16

2、在项目目录下新建一个 output\vgg16\voc_2007_trainval+voc_2012_trainval\default路径，并将default\voc_2007_trainval\default文件夹中的文件全部拷贝过去。
3、运行 python demo.py