Bug处理:YoloWorld的训练过程中进行验证出现IndexError: list index out of range

在YoloWorld的训练中,进行周期验证的时候报出该错误:

File "/xx/xx/anaconda3/envs/yoloworld/lib/python3.8/site-packages/mmdet/evaluation/metrics/coco_metric.py", line 243, in results2json
data['category_id'] = self.cat_ids[label]
IndexError: list index out of range

Issue中作者的建议是仿照CocoDataset创建一个自己的Dataset类。具体地就得自己研究尝试了,思路是先建立BomoDataset和继承前者的YoloWorldBomoDataset(名字自己随便取的),再在各自的__init__.py中引入。


具体步骤如下:

Step1. 在anaconda3/envs/yoloworld/lib/python3.8/site-packages/mmdet/datasets/coco.py中定义自己的BomoDataset类。

复制class CocoDataset并命名为自己的class BomoDataset,然后修改变量METAINFO中的类别名。如:

METAINFO = {
        'classes':
        ("Big black spot", ..),
    	..

anaconda3/envs/yoloworld/lib/python3.8/site-packages/mmdet/datasets/__init__.py中引入BomoDataset

from .coco import BomoDataset
__all__ = [.., 'BomoDataset']

Step2. 在anaconda3/envs/yoloworld/lib/python3.8/site-packages/mmyolo/datasets/yolov5_coco.py中定义自己的YoloWorldBomoDataset类。

复制class YOLOv5CocoDataset并命名为自己的class YoloWorldBomoDataset(BatchShapePolicyDataset, BomoDataset),注意继承。

anaconda3/envs/yoloworld/lib/python3.8/site-packages/mmyolo/datasets/__init__.py中引入YoloWorldBomoDataset

from .yolov5_coco import YoloWorldBomoDataset
__all__ = [.., 'YoloWorldBomoDataset']

Step3. 在configs/finetune_bomo/yolo_world_l_dual_vlpan_2e-4_80e_8gpus_finetune_bomo.py中,将train和valid的dataloader修改为自己新定义的YoloWorldBomoDataset

dataset=dict(
        ..
        dataset=dict(
            type='YoloWorldBomoDataset
           	..

到这里我就解决该问题了。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值