windows10下TensorFlow版本Faster RCNN复现
目录
-
1、模型准备工作
-
- 1-1 模型运行环境
- 1-2 模型地址
- 1-3 数据集及网络地址
-
2、模型训练
-
- 2-1 train前准备
- 2-2 下载vgg16及resnet模型
- 2-3 train
1、模型准备工作
1-1 模型运行环境
python3.6
TensorFlow1.12-GPU
windows10
Ananonda
Pycharm
CUDA9+cuDNN7.6
opencv-python
tensorflow-plot
cython
easydict
Pillow
matplotlib
scipy
1-2 模型地址
1-3 数据集及网络地址
VOC2007：https://uinedu-my.sharepoint.com/:f:/g/personal/19604_myoffice_site/EiLTzAbNirROrQQF20eupMQB-KpIfZOa7w2YS5MB2ARvSA
http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz
resnet50_v1:http://download.tensorflow.org/models/resnet_v1_50_2016_08_28.tar.gz
resnet101_v1:http://download.tensorflow.org/models/resnet_v1_101_2016_08_28.tar.gz
2、模型训练
2-1 train前准备
打开cmd，激活TensorFlow-GPU环境，并转入模型中的下述路径
D:\pytrain\my_models\Faster-RCNN-TensorFlow-Python3-master\data\coco\PythonAPI
运行命令
python setup.py build_ext --inplace python setup.py build_ext --install再转到模型文件中的这个路径
D:\pytrain\my_models\Faster-RCNN-TensorFlow-Python3-master\lib\utils
运行命令
python setup.py build_ext --inplace当上述三个命令运行完后，会发现该文件夹下多个红框中的文件
D:\pytrain\my_models\Faster-RCNN-TensorFlow-Python3-master\data\coco\PythonAPI\pycocotools
该文件夹中会多出红框文件
D:\pytrain\my_models\Faster-RCNN-TensorFlow-Python3-master\lib\utils
随后第一部分的链接中下载公开VOC数据集，将其放在下述路径
D:\pytrain\my_models\Faster-RCNN-TensorFlow-Python3-master\data\VOCdevkit2007\VOC2007
2-2 下载vgg16及resnet模型
将下载好的模型，放入下述文件夹中
D:\pytrain\my_models\Faster-RCNN-TensorFlow-Python3-master\data\imagenet_weights
其中imagenet_weights文件夹需要自己创建
2-3 train
运行train.py即可进行训练