使用自己的数据集,格式为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,
))