Faster rcnn排坑指南


1、在全球最大攪基論壇下載Faster rcnn代碼(代碼基於3.5,但我的環境為py3.6):https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.5

2、修改Faster-RCNN-TensorFlow-Python3.5-master\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=[], )

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

python setup.py build_ext --inplace
生成cython_bbox.c和cython_bbox.pyx,然后將這兩個改名為bbox.c和bbox.pyx。

 再在./data/coco/PythonAPI下面運行

 python setup.py build_ext --inplace

 python setup.py build_ext install

下載數據集

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目錄下

下載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\目錄下

都弄好以后,就可以訓練了執行:

python train.py

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM