yolov5中的误判

Question

我用 yolo5 训练了一个单类模型。包含 400 张图像的数据集分为训练、验证和测试数据集。这是训练代码train code 的笔记本。 train 得到了很好的结果。但是，当我用导出的模型构建实时对象检测应用程序时，结果并不理想，它可以按预期识别对象。但是检测中有很多误判。检测结果： result

结果中只有一个百事可乐，为什么时间戳和其他小区域都被识别为百事可乐。

谁能告诉我结果的原因？

Answer 1

您的问题缺乏很多细节，但我猜您的图像训练不足。 400 只是一个玩具，很可能导致过拟合。

您将需要查看 COCO 或 ImageNet 之类的数据集以获得合理概括的结果。