本博文介绍如何在Caffe上实现R-FCN,关于R-FN的算法讲解和Caffe代码详解可以参考另外一篇博客:Object Detection算法——R-FCN算法及Caffe代码详解
git地址:https://github.com/Orpine/py-R-FCN
基本上按照这个git上的readme.md的流程进行并假设你本机的Caffe本来就可以正常使用。本篇博文主要包括:搭好项目环境,运行demo,自己训练R-FCN模型。接下来详细讲解可能会遇到的坑:
一、搭好项目环境
1、先从git上拉取R-FCN
git clone https://github.com/Orpine/py-R-FCN.git
并假设拉取后的文件名是 RFCN_ROOT
2、拉取Caffe
cd $RFCN_ROOT
git clone https://github.com/Microsoft/caffe.git
3、编译Cython
cd $RFCN_ROOT/lib
make
4、复制Makefile.config(这一步git上面漏掉了)
cd $RFCN_ROOT/caffe
cp Makefile.config.example Makefile.config
5、编译Caffe和pycaffe
cd $RFCN_ROOT/caffe
make -j32 && make pycaffe
会报这个错:
原因是在make -j32的时候找不到hdf5文件。解决办法就是修改Makefile.config文件,将Makefile.config中的下面这两行:
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS :=