py-r-fcn训练自己的数据
按照voc格式制作训练数据,Main里面只需要train,trainval就可以。文件夹格式和voc2007的要一样,这样可以防止更改里面的东西。
以及.prototxt
在编译caffe时如果报cudnn错误(版本不匹配),注释掉cudnn时可以正常编译时,则将src/caffe/layers/,src/caffe/util,include/caffe/layers/,include/caffe/util/下面的cudnn_更换成可以编译通过的版本
r-fcn 下面的lib需要make
python ./tools/train_net.py --gpu 1 --solver models/pascal_voc/ResNet-50/rfcn_end2end/solver_ohem.prototxt --weights data/models/ResNet-50-model.caffemodel --iters 3000 --cfg experiments/cfgs/rfcn_end2end_ohem.yml --imdb voc_2007_trainval
参考:https://blog.****.net/byplane/article/details/73603861