machine-learning - yolov5中的误判

Question

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

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

有人可以告诉我结果的原因吗？

标签： machine-learningyoloyolov5

Answer 1

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

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