首先是从官网克隆代码:
git clone https://github.com/xiaolonw/adversarial-frcnn
得到源码之后,要注意的是还要克隆fast RCNN的源码
git clone https://github.com/rbgirshick/fast-rcnn(运行代码需要里面的ss算法部分)
在运行作者的源码之前,要准备安装caffe,caffe安装请参照网上博客,不同的人会出现不同的问题,安装完之后记得添加到环境变量
,不然会出错。训练代码前需要下载VOC数据集,http://pjreddie.com/media/files/VOCtest_06-Nov-2007.tar
和http://pjreddie.com/media/files/VOCtrainval_06-Nov-2007.tar。下载完之后放入指定的文件中然后就可以运行训练代码了。
然后cd到代码的文件夹中,使用 sh train.sh命令来运行训练过程,整个代码只需要运行这么一句就行了。但是这个时候会提示一个
有关utils.cython_bbox模块的错误,提示没有这个模块。这时候的解决办法就是cd到/lib文件夹中,将setup.py文件中的
utils.cython_bbox替换成utils.bbox,然后make,然后依次执行Python setup.py build, Python setup.py install
如果还是出现cython_bbox的错误,直接将出错误的文件中的cython_bbox改成bbox就行。
最后再执行sh train.sh文件进行训练。目前我的代码证在跑,由于电脑内存不够出现核心已转储的错误。