1、训练平台:R5 3600、RTX2060Super,16G运行内存。
2、源码地址:https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3
3、使用git下载源码,git clone https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.git
![](https://img-blog.csdnimg.cn/20200324172220325.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQzNzgyMQ==,size_16,color_FFFFFF,t_70)
4、如果直接运行会出现以下错误
解决请参考我上一章博客:https://blog.csdn.net/weixin_40437821/article/details/105075571
(4.1)修改Faster-RCNN-TensorFlow-Python3\data/coco/PythonAPI/setup.py文件:在第15行加上
,
Extension( 'lib.utils.cython_bbox',
sources=['../../../lib/utils/bbox.c','../../../lib/utils/bbox.pyx'],
include_dirs = [np.get_include(), '/lib/utils'],
extra_compile_args=[], )
(4.2)由于没有bbox.c和blob.py文件。所以要先在Faster-RCNN-TensorFlow-Python3\lib\utils执行
python setup.py build_ext --inplace
生成cython_bbox.c和cython_bbox.pyx,然后将这两个改名为bbox.c和bbox.pyx。如下图所示
(4.3) 再在./data/coco/PythonAPI下面运行
python setup.py build_ext --inplace
python setup.py build_ext install
5、下载数据集
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
将三个文件同时解压到VOCdevkit2007目录下,并将VOCdevkit2007目录存放于data目录下(建议用WinRAR解压到指定目录,会自动合并文件夹)
6、下载VGG模型:http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz
解压并改名为vgg16.ckpt(原来是vgg_16.ckpt)后存放在Faster-RCNN-TensorFlow-Python3.5-master\data\imagenet_weights\
目录下。(PS:需要新建文件夹imagenet_weights
)
7、运行train.py即可