使用 YOLO v5 训练自定义对象检测模型

Posted 2023-03-12

【中文标题】使用 YOLO v5 训练自定义对象检测模型

【英文标题】：Train A Custom Object Detection Model with YOLO v5

【发布时间】：2020-12-12 16:19:16

【问题描述】：

我正在尝试使用 Yolo v5 训练模型以检测销售传单上的多个对象。训练中使用的数据集中的每张图像只包含一个对象，并且显然是一个边界框。 我想知道这是否会以不好的方式影响模型的性能？因为我最后要做的是检测每个销售传单上的多个对象。 感谢您的帮助。

【参考方案1】：

如果你这样工作，它可能会降低你的 AP，但试试看。这实际上取决于您的训练图像和数据增强。我不知道 YOLOv5，但 YOLOv4 有它的 Mosaic Data Augmentation，我猜它会以某种方式解决你的问题。