按照voc格式制作训练数据,Main里面只需要train,trainval就可以。文件夹格式和voc2007的要一样,这样可以防止更改里面的东西。
py-r-fcn训练自己的数据
py-r-fcn训练自己的数据
以及.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.csdn.net/byplane/article/details/73603861

相关文章:

  • 2021-09-13
  • 2021-11-21
  • 2021-11-29
  • 2021-04-15
  • 2022-01-22
  • 2022-02-02
猜你喜欢
  • 2021-12-11
  • 2022-12-23
  • 2021-07-15
  • 2021-09-07
  • 2021-10-18
  • 2022-01-25
  • 2021-07-22
相关资源
相似解决方案