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

Question

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

Answer 1

如果您像这样工作，它可能会降低您的AP，但请尝试一下。这实际上取决于您的训练图像和数据扩充。我不了解YOLOv5，但YOLOv4具有其镶嵌数据增强功能，可以以我想的方式解决您的问题。