使用mmdetection训练自己voc格式的数据集

本文档详细介绍了如何使用mmdetection框架训练一个包含9类的自定义VOC格式数据集。首先,需要在mmdet/datasets/init.py中注册新数据集,并在voc.py中修改以适应自定义类别。接着,将数据放置在名为VOC2012的文件夹中,包括JEPGImages和Annotations。然后,更新配置文件中的数据路径以指向新数据集。此外,文中提到还可以通过offline conversion转换数据,但因为需要频繁查看图像和标注,作者选择了online conversion方法。
摘要由CSDN通过智能技术生成

使用mmdetection训练自己的数据集

之前使用mmdetection训练的时候,找过几个老哥的博客,但都只是布置voc07或者12的数据训练,没有定义自己的数据集,所以这里特地把训练自己数据集的方式记录下来

场景:【有voc形式的数据集,类别数和名称自定】
我有9类的数据集 名为 “MyDataset”
类别名为 [‘000’,‘001’,‘002’,‘003’,‘004’,‘005’,‘006’,‘007’,‘008’]

mmdetection的最新版本(2019/7/4查看)的GETTING_STARTED.md 里面已经给出了两种使用自己数据集的方法:
两种定义数据集方法
为了符合之前自己的习惯,选择用online conversion,自己用labelImg标注的xml文件和图像共同输入
下面是修改方法

  1. 在mmdet/datasets/init.py 里注册自己的数据集
    在以下两个地方添加

    注册自己的数据集
  2. mmdet/datasets 里有voc.py里面存放了voc数据集的信息,我们改成自己
  • 2
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值