CaffePython版本Faster R-CNN+ZF代码运行

Posted 糖梦梦是女侠

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了CaffePython版本Faster R-CNN+ZF代码运行相关的知识,希望对你有一定的参考价值。

Python Github代码地址:https://github.com/rbgirshick/py-faster-rcnn

相关参考博客:

【1】http://blog.csdn.net/sinat_30071459/article/details/51332084

【2】http://www.cnblogs.com/CarryPotMan/p/5390336.html

【3】http://blog.csdn.net/samylee/article/details/51201744


下面是我自己在跑python版本Faster R-CNN代码ZF模型的流程:

1.git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git

2.进入py-faster-rcnn/lib  make

3.将make.config文件复制到py-faster-config/caffe-fast-rcnn中

   make.config文件及后面用到的VOC数据集及与训练模型imagenet下载地址为:http://pan.baidu.com/s/1slGgac9

4.在py-faster-config/caffe-fast-rcnn下

    make all

    make test (可不要)

    make runtest (可不要)

    make pycaffe


5.复制数据集VOCdevkit2007及与训练模型imagenet_modelst到py-faster-rcnn/data下

   或者 cd py-faster-rcnn下:  ./data/scripts/fetch_faster_rcnn_models.sh


6.在py-faster-rcnn/lib/fast-rcnn/train.py中增加/py-faster-rcnn/caffe-fast-rcnn/python的绝对路径:


7.更改迭代次数

   在py-faster-rcnn/tools/train_faster_rcnn_alt_opt.py中更改(这里我将原来的迭代次数缩小了100倍):



8.可能会出现内存溢出的问题,需要py-faster-rcnn/lib/faster-rcnn/config.py中的图片大小,论文中将最短边缩放到600 pixels,一般电脑吃不消,可以改为450 pixels,对应MAX_SIZE改为750,如下:



9.开始训练

在py-faster-rcnn下执行  ./experiments/scripts/faster_rcnn_alt_opt.sh 0 ZF pascal_voc

    

以上是关于CaffePython版本Faster R-CNN+ZF代码运行的主要内容,如果未能解决你的问题,请参考以下文章

用pycharm训练Faster R-CNN,遇到如图问题,怎么办?

Widows下Faster R-CNN的MATALB配置(GPU)

faster-rcnn算法总结

MXNet的Faster R-CNN(基于区域提议网络的实时目标检测)《8》

MXNet的Faster R-CNN(基于区域提议网络的实时目标检测)《9》

Faster R-CNN简介及其发展