其他分享
首页 > 其他分享> > Faster rcnn排坑指南

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

我这里出现错误:UnicodeEncodeError: 'utf-8' codec can't encode character '\udcbe' in position 151: surrogates not allowed

解决办法:

 

标签:Faster,..,utils,py,bbox,lib,rcnn,排坑
来源: https://www.cnblogs.com/roscangjie/p/10945001.html