YOLOv5 目标检测模型

Question

我在训练 yolov5 时遇到问题。 我可以训练具有多个类的模型，例如 ['cat','dog','rabbit']。 我想为这个模型再添加一个类，['snake']，这样我的模型可以一次检测到所有四个。 我尝试过模型集成和冻结层，但这些并没有给我想要的结果。 请谁能帮我弄清楚如何做到这一点。 提前致谢

Answer 1

我想帮助解决您的问题，但信息太少。 出现此问题通常是由于以下原因：

1. 没有为 ['snake'] 做标签
2. ['snake'] 没有添加到 yaml 文件的类中
3. 训练数据集小或标注不正确。

可能还有其他原因，但是如果没有运行detect.py的结果图片等信息，很难提供详细的帮助。