edgeyolo icon indicating copy to clipboard operation
edgeyolo copied to clipboard

请问为什么evaluate时会报list out of range

Open ysyydsysyyds opened this issue 2 years ago • 1 comments

我是在将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

ysyydsysyyds avatar Jul 10 '23 07:07 ysyydsysyyds

coco原数据集能跑起来没问题说明dataloader代码没问题,可能这个代码对数据集的规范性的要求比较高,实在不行的话就转yolo格式吧。

LSH9832 avatar Dec 06 '23 05:12 LSH9832