mmdetection3.0.0用coco训练正常，voc格式训练出现map很低的情况

最新推荐文章于 2024-02-06 15:27:00 发布

Pliter

最新推荐文章于 2024-02-06 15:27:00 发布

阅读量338

点赞数 2

文章标签：人工智能算法机器学习

本文链接：https://blog.csdn.net/Pliter/article/details/134389961

版权

记录下遇到的问题。训练SSDD数据集的时候，发现coco格式训练正常，但voc格式训练出现map值很低，一直升不上去。试了下2.x版本的mmdetection训练voc格式没问题，解决方案是在configs/_base_/datasets/voc0712.py中删掉bbox_min_size=32即可

train_dataloader = dict(
    batch_size=2,
    num_workers=2,
    persistent_workers=True,
    sampler=dict(type='DefaultSampler', shuffle=True),
    batch_sampler=dict(type='AspectRatioBatchSampler'),
    dataset=dict(
        type='RepeatDataset',
        times=1,
        dataset=dict(
                type=dataset_type,
                data_root=data_root,
                ann_file='VOC2007/ImageSets/Main/trainval.txt',
                data_prefix=dict(sub_data_root='VOC2007/'),
                filter_cfg=dict(
                        filter_empty_gt=True, min_size=32,
                        # bbox_min_size=32                #####################
                                    ),
                    pipeline=train_pipeline,
                    backend_args=backend_args),
 
            ))