Faster-RCNN+TensorFlow 詳細訓練過程（附github源碼）

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

​項目整體代碼結構

4、如果直接運行會出現以下錯誤

​​解決請參考我上一章博客：https://blog.csdn.net/weixin_40437821/article/details/105075571

​（4.1）修改Faster-RCNN-TensorFlow-Python3\\data/coco/PythonAPI/setup.py文件：在第15行加上

<code>,
    Extension( 'lib.utils.cython_bbox',
               sources=['../../../lib/utils/bbox.c','../../../lib/utils/bbox.pyx'],
               include_dirs = [np.get_include(), '/lib/utils'], 
               extra_compile_args=[], )/<code>

（4.2）由於沒有bbox.c和blob.py文件。所以要先在Faster-RCNN-TensorFlow-Python3\\lib\\\\utils執行

<code>python setup.py build_ext --inplace/<code>

生成cython_bbox.c和cython_bbox.pyx，然後將這兩個改名為bbox.c和bbox.pyx。如下圖所示

​​(4.3) 再在./data/coco/PythonAPI下面運行

<code>python setup.py build_ext --inplace
python setup.py build_ext install/<code>

5、下載數據集

<code>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/<code>

將三個文件同時解壓到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即可

​​

閱讀更多 AI機器學習與數據挖掘 的文章

關鍵字: 源碼 WinRAR Pascal