[Bug]:

[asylum127]

Before Reporting

• [X] I have pulled the latest code of main branch to run again and the bug still existed. 我已经拉取了主分支上最新的代码，重新运行之后，问题仍不能解决。

• [X] I have read the README carefully and no error occured during the installation process. (Otherwise, we recommand that you can ask a question using the Question template) 我已经仔细阅读了README上的操作指引，并且在安装过程中没有错误发生。（否则，我们建议您使用Question模板向我们进行提问）

Search before reporting

• [X] I have searched the DAMO-YOLO issues and found no similar bugs. 我已经在issue列表中搜索但是没有发现类似的bug报告。

OS

ubuntu

Device

nvidia 2080ti

CUDA version

cuda10.1

TensorRT version

No response

Python version

3.7

PyTorch version

1.7.2

torchvision version

0.8.0

Describe the bug

我在训练自己的数据集后，在推理时出现报错 数据集中标签总共有三个类，推理结果中类为“1，2，3”（cls_inds为‘1，2，3’，而非‘0，1，2） 在“visualize.py”中直接用"cls_ids"作为索引会导致报错

To Reproduce

我将代码改为： 可以正常推理 请问这个bug是原来代码中就存在的问题，还是我训练时的出错导致的

Hyper-parameters/Configs

No response

Logs

No response

Screenshots

No response

Additional

No response

[XianzheXu]

请问您使用的是最新的代码吗？如果不是的话麻烦您pull一下最新代码再尝试推理。如果是的话，麻烦查看一下训练和推理的config文件，其中head的num_classes是否是正确设置的？

[asylum127]

请问您使用的是最新的代码吗？如果不是的话麻烦您pull一下最新代码再尝试推理。如果是的话，麻烦查看一下训练和推理的config文件，其中head的num_classes是否是正确设置的？

num_classes设置为3，我的数据中有三个类的标签

[XianzheXu]

您的代码是最新的吗？我们在commit: ee8050e1621897b5ec84670a1b33b6a086a75b55中修复过此问题，麻烦再确认一下您当前代码是否包含该次commit. 要查看代码commit记录，可以在DAMO-YOLO目录下使用git log 命令。