R-FCN+ResNet-50用自己的数据集训练模型(python版本)

说明：

本文假设你已经做好数据集，格式和VOC2007一致，并且Linux系统已经配置好caffe所需环境（博客里教程很多），下面是训练的一些修改。

py-R-FCN源码下载地址：

https://github.com/Orpine/py-R-FCN

也有Matlab版本：

https://github.com/daijifeng001/R-FCN

本文用到的是python版本。

本文主要参考https://github.com/Orpine/py-R-FCN。

准备工作：

（1）配置caffe环境(网上找教程)

（2）安装cython, python-opencv, easydict

pip install cython
pip install easydict
apt-get install python-opencv

然后，我们就可以开始配置R-FCN了。

1.下载py-R-FCN

git clone https://github.com/Orpine/py-R-FCN.git

下面称你的py-R-FCN路径为RFCN_ROOT.

2.下载caffe

注意，该caffe版本是微软版本

cd $RFCN_ROOT
git clone https://github.com/Microsoft/caffe.git

如果一切正常的话，python代码会自动添加环境变量 $RFCN_ROOT/caffe/python，否则，你需要自己添加环境变量。

3.Build Cython

cd $RFCN_ROOT/lib
make

4.Build caffe和pycaffe

cd $RFCN_ROOT/caffe
cp Makefile.config.example Makefile.config

然后修改Makefile.config。caffe必须支持python层，所以WITH_PYTHON_LAYER := 1是必须的。其他配置可参考： Makefile.config

接着：

cd $RFCN_ROOT/caffe
make -j8 && make pycaffe

如果没有出错，则：

5.测试Demo

经过上面的工作，我们可以测试一下是否可以正常运行。

我们需要下载作者训练好的模型，地址：链接：http://pan.baidu.com/s/1kVGy8DL<