faster_rcnn编译及训练

faster_rcnn编译时需要替换的文件:

第一部分:

caffe-master/include/caffe/util 下面的cudnn.hpp

以及 caffe-my/src/caffe/util 对应的cpp 文件:cudnn.cpp

第二部分:caffe-master/include/caffe/layers下面的hpp文件


第三部分:caffe-master/src/caffe/layers样的对应的cpp文件和cu文件:

按照正常的cmake以及make    make -j8 && make pycaffe 




faster_rcnn训练过程:

一种是end-to-end,在终端执行:

旋转GPU-0,网络为:ZF  数据集为:pascal_voc

./experiments/scripts/faster_rcnn_alt_opt.sh 0 ZF pascal_voc






猜你喜欢

转载自blog.csdn.net/Lisatiy/article/details/80646092
今日推荐