Yolov5 模型捕获除预期或期望对象之外的其他对象

Question

问题

我有一个 YOLO V5 m 模型用于数据集来检测某种产品，但在不同情况和场景以及各种照明条件下训练该产品的各种图片的数据集之后，在 50 个时期和不重复的情况下Yolo v5 中等架构，批量大小为 16。我注意到该模型依赖于颜色特征，这意味着检测到的对象具有黄色及其阴影。我使用了 yolo github 克隆附带的 train.py。我应该怎么做才能解决这个问题？请帮助我，也请在下面找到用于训练的数据集（大约 450 张图片）。

需要检测的产品：

检测到的其他产品：

Google Drive 中用于训练的数据集（大约 450 张图像）：

https://drive.google.com/file/d/1oBH28fbyhEmabLLx2bPXpEMQ0s2zhg__/view?usp=sharing

Answer 1

也许您的数据集太小。您可以尝试使用增强来扩展您的数据集。