我正在训练 yolov5 模型来对汽车 4 个不同部分(底盘、前扰流板、轮毂盖和车轮)的图像进行分类,但它的猜测是完全错误的,它无法区分底盘和前扰流板以及车轮和车轮分别是轮毂盖。对于 100 和 1000 个 epoch 的训练都是如此。谁能告诉我可能出了什么问题吗?
如果不太了解您训练模型所依据的数据量,并且仅基于视觉结果,我确信这还不够。 Yolov5 建议每个类别使用 10000 个实例以获得良好的稳健结果。增加训练模型的数据量。
低置信度反映了您的模型在识别每个对象时的不确定性。您检测到的所有对象都没有很强的置信度分数。