py-r-fcn训练自己的数据

本文介绍如何使用Py-R-FCN项目搭建基于VOC格式的目标检测训练环境,并提供详细的步骤指导,包括数据集准备、Caffe编译配置、训练脚本运行等关键环节。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

按照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.youkuaiyun.com/byplane/article/details/73603861
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值