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