mmdetection训练自己的模型

参考文档:
mmdetection入门文档

预训练权重下载model_zoo.md

>>训练自己的模型

1.修改配置文件

位置:mmdetection/configs/__ base__/models/faster_xxxx.py
操作:

  • num_class=N
  • dataset_type = ‘VOCDataset’ ;data_root = ‘data/VOCdevkit/’
  • ann_file = data_root + ‘VOC2007/ImageSets/Main/xx.txt’
  • img_prefix = data_root + 'VOC2007/'等

说明:N表示分类类别,不包含背景,2.x版本中都不需要+1

(可选)修改输入图片大小
位置:mmdetection/configs/__base __/datasets/voc0712.py
操作:img_scale=(1000, 600)
说明:如果电脑跑不动,可以改小一点

2.1修改标签1
位置:mmdetection/mmdet/core/evaluation/class_names.py
操作:根据格式修改标签即可,以VOC为例
说明:

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值