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

最新推荐文章于 2024-11-23 23:00:30 发布

仙女修炼史

最新推荐文章于 2024-11-23 23:00:30 发布

阅读量4.7k

点赞数 4

分类专栏：实践文章标签：目标检测深度学习计算机视觉

本文链接：https://blog.youkuaiyun.com/weixin_45209433/article/details/121749590

版权

实践专栏收录该内容

22 篇文章

订阅专栏

这篇博客详细介绍了如何使用SSD300模型训练自定义的VOC格式数据集。主要步骤包括修改配置文件（如`voc0712.py`和`xml_style.py`）以适配数据集路径，调整类别数和batch size，以及修改学习率策略。此外，还涉及到了处理PNG图像格式和计算mAP时的注意事项。

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

下面的例子是使用ssd300训练自己的数据集（voc格式）

一、configs/base/datasets/voc0712.py

修改data_root为自己数据集的root
所有的img_prefix，都修改为data_root，不需要加’2007’或者’2012’，这里不修改也可以，需要把自己的数据放到一个叫2007或者2012的文件夹中。修改之后，就可以随意的放置自己的数据集了。
mmdet/datasets/xml_style.py 中，需要使用self.img_prefix来找到xml正确的路径，进而解析xml，因此img_prefix = data_root需要这样修改。
type=‘RepeatDataset’ 是方便增加数据集，如果是新增加的数据，可以写成这种形式 ann_file = [a, b], img_prefix = [a, b]

data = dict(
    samples_per_gpu=2,
    workers_per_gpu=2,
    train=dict(
        type='RepeatDataset',
        times=3,
        dataset=dict(
            type=dataset_type,
            ann_file = data_root + '/ImageSets/Main/train.txt',
            img_prefix = data_root,
            pipeline=train_pipeline)),

    val=dict(
        type=dataset_type,
        ann_file=data_root + '/ImageSets/Main/test.txt',
        img_prefix=data_root,
        pipeline=test_pipeline),
    test=dict(
        type=dataset_type,
        ann_file=data_root + '/ImageSets/Main/test.txt',
        img_prefix=data_root,
        pipeline=test_pipeline))

二、configs/pascal_voc/ssd300_voc0712.py

第7行：修改num_classes成自己的类别数，如果是一个类别，需要写成CLASSES = (person,)，否则会出现 “AssertionError: CLASSES in RepeatDatasetshould be a tuple of str.Add comma if number of classes is 1 as CLASSES = (person,)” 的错误。
第49行：修改samples_per_gpu，就是设置batch_size
第56行：修改optmizer的学习率大小，以及学习率策略

三、mmdet/datasets/voc.py 中：

CLASSES 改为自己的类别
由于在第一步中，我们舍弃了2007或者2012，因此这里的代码需要修改一下，25、26行的代码直接不要，这样会把数据集img_prefix限制在2007或者2008

else:
raise ValueError('Cannot infer dataset year from img_prefix')

3、计算mAP的时候直接使用’voc07’

66到69的代码
 if self.year == 2007:
    ds_name = 'voc07'
 else:
    ds_name = self.CLASSES

直接改为：
ds_name = 'voc07'

四、mmdet/core/evaluation/class_names.py

voc_classes() 改为自己的类别名字

五、如果自己的数据集是png文件，如何修改

在/mmdet/datasets/xml_style.py 51行 .jpg 可以改为.png