新版mmdetection训练自己的数据集

配置

最好的情况是按照mmdetection的文档介绍,一步一步走。pytorch1.5没有配置成功。pytorch1.3基本可以很顺利的配置好,但是要注意的是cuda的版本问题。

voc数据集

用labelimg或其他工具标注好的voc数据集,按照mmdetection中文档的介绍的格式放好就行。
以上两个步骤不是最主要的。

训练

选取好自己想用的head模型,比如faster rcnn或者retinanet。因为mmdetection经历了一次大的更新,很多地方都有了变动。
1.关于文件路径 和class类别数量的问题,都去configs/__base __/文件夹下去设置。
datasets/voc0712.py设置路径
models/.py设置类别数量(注意:以前的版本都是类别数+1,新版只需要类别数就行,比如训练1类就是num_class=1不然回报list out of range错误)
schedule/
.py设置epochs
2.设置好之后去mmdet/datasets/voc.py去设置类别名,如果是1类的话一定要加个逗号(‘smoke’,)
3.去mmdet/core/evaluation/class_name.py设置类别名,可以不用加逗号
4.学习率去configs/特定head.py下去设置 同事configs的base下的schedule也要更改
这些都设置好之后就可以开始训练了用官方给的运行命令即可。

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值