mmdetection 测试时报错 data[‘category_id‘] = self.cat_ids[label] IndexError: list index out of range

使用自己的数据集,格式为coco格式,类别和coco.py里面不一致导致报错

修改自己的config文件,在test_dataloader的dataset里面加上metainfo,设置为自己的类别即可

# Modify dataset related settings
metainfo = {
    'classes': ('Inlet', 'Slightshort', 'Generalshort', 'Severeshort', 'Outlet' ),
}
test_dataloader = dict(
    batch_size=1,
    num_workers=2,
    persistent_workers=True,
    drop_last=False,
    sampler=dict(type='DefaultSampler', shuffle=False),
    dataset=dict(
        type=dataset_type,
        data_root=data_root,
        metainfo=dict(classes=classes),    # 添加这一行设置类别,为上面的metainfo
        ann_file=data_root + 'annotations/instances_test2017.json',
        data_prefix=dict(img='test2017/'),
        test_mode=True,
        ))

猜你喜欢

转载自blog.csdn.net/ZZZZ_Y_/article/details/134408415
今日推荐