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

Question

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

Answer 1

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