基于tensorflow+python3.5环境实现Faster R-CNN

1.程序源代码下载

https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.5

该程序是针对windows环境写的,所以在ubuntu中配置时需要做一些修改,详见以下步骤。

2.安装所依赖的库

按照源程序的readme.md文件,安装以下库:

pip install cythoncython opencv-opencv easydict

后来在运行过程中numpy相关的函数报错,所以又重新安装了numpy

3. 更换cython_bbox文件

如果不进行这步操作将会提示:ImportError: No module named 'lib.utils.cython_bbox'

转到.\lib\utils目录,下载cython_bbox.cpython-35m-x86_64-linux-gnu.so存放至该目录,并删除该目录下的cython_bbox.cp35-win_amd64.pyd文件。

下载地址:https://pan.baidu.com/s/1cHSqnC_PIttwcjrILHapRQ

密码:q3hz

4.下载数据

wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar

在./data目录现建立一个文件夹,命名为VOCdevkit2007,将上述文件全部解压到该文件夹中,解压后VOCdevkit2007文件夹中应该有三个文件:VOC2007,VOCcode,VOCdevkit

5.下载pre-trained VGG16

http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz

其存放路径为.\data\imagenet_weights\vgg16.ckpt

6.转到.\data\coco\PythonAPI目录

分别运行:

python setup.py build_ext --inplace
python setup.py build_ext install

7.Run train.py

猜你喜欢

转载自blog.csdn.net/wangxinsheng0901/article/details/81805372