关于plot_images的索引超出范围的问题

[LaoGanGit]

作者你好，我在运行我自己的数据集，到了最后一个epoch时，出现了下面的错误
Traceback (most recent call last):
File "F:/jinyuan/yolor-main/train.py", line 537, in
train(hyp, opt, device, tb_writer, wandb)
File "F:/jinyuan/yolor-main/train.py", line 344, in train
log_imgs=opt.log_imgs if wandb else 0)
File "F:\jinyuan\yolor-main\test.py", line 226, in test
plot_images(img, output_to_target(output, width, height), paths, f, names) # predictions
File "F:\jinyuan\yolor-main\utils\plots.py", line 163, in plot_images
cls = names[cls] if names else cls
IndexError: list index out of range
因为这个问题，导致最后在run里面少了很多图片，没有mAp的曲线图

请教，这是什么原因导致的呢，有何解决的方法，谢谢

[WongKinYiu]

https://github.com/WongKinYiu/yolor/blob/main/test.py#L312
這裡要改成您的names fiile

[LaoGanGit]

https://github.com/WongKinYiu/yolor/blob/main/test.py#L312
這裡要改成您的names fiile

好的，非常感谢您的解答

[LaoGanGit]

https://github.com/WongKinYiu/yolor/blob/main/test.py#L312
這裡要改成您的names fiile
作者您好，我按您的建议修改了下还是出现原来的错误，您看下我这个自定的names.file和yaml有问题吗？
是否除了test.py的names修改外还有哪些跟着修改呢？
另外detect.py#176的names我也修改成跟test.py一样的路径和内容了

[Hu-nie]

hello

I have the same problem. Did you figure it out?

[Hu-nie]

I solved this problem.

#17

[amadeusneko]

请问你解决掉这个问题了吗？我也遇到了相同的问题

[francescotaioli]

I also have this problem even if I create a custom data.names file with the correct label

[x-yy0]

I have the same problem. Did you figure it out?

[WenTianWuHuo]

You need to change the. cfg file you are using, changing 'classes=80' to 'classes=yours classes' and' filters=225 'to' filters=(5+yours classes) * 3 '.