A-Fast-RCNN算法来自论文:A-Fast-RCNN: Hard Positive Generation via Adversary for Object Detection。
项目的git地址:https://github.com/xiaolonw/adversarial-frcnn
因为git上对于如何训练模型,如何测试数据介绍得比较简单,因此自己亲自试验了一下,并记录下过程。
1、从Faster RCNN的git上拉取项目:
git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git
假设拉取下来后的文件名为$FRCN_ROOT
2、从A-Fast-RCNN的git上拉取项目
从A-Fast-RCNN上下载项目,然后把这个项目里的所有东西手动覆盖到$FRCN_ROOT这个目录下的所有文件。
git clone --recursive https://github.com/xiaolonw/adversarial-frcnn
3、复制配置文件并编译caffe和pycaffe:
cd $FRCN_ROOT/caffe-fast-rcnn
cp Makefile.config.example Makefile.config
这个Makefile.config需要更改两处地方:
一处是将makefile.config里面的以下两行:
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
换成这两行:其实就是加上了hdf5的路径
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/
LIBRARY_DIRS