edgeyolo
edgeyolo copied to clipboard
请问为什么evaluate时会报list out of range
我是在将edgeyolo用于自己的数据集时遇到这个问题的,如果直接训练的话可以完成训练,但在评估时报此错误。之后直接评估也是此错误,具体报错信息如下。我的数据集是coco格式的,也可以用别的模型跑起来。不过我的类别只有四个,是否是类别数量太少的原因?该如何修改呢? File "/home/workspace/edgeyolo/edgeyolo/train/val/coco_evaluator.py", line 192, in convert_to_coco_format label = self.dataloader.dataset.class_ids[int(cls[ind])] IndexError: list index out of range
coco原数据集能跑起来没问题说明dataloader代码没问题,可能这个代码对数据集的规范性的要求比较高,实在不行的话就转yolo格式吧。