如何使用 XML 注释可视化数据集并将其用于 YOLOv5

How to visualize dataset with XML Annotation and use it for YOLOv5

我正在为 YOLOv5 进行自定义训练。我想用来自 UA-DETRAC 的数据集训练 YOLOv5,其中包含 XML 包含一些图像的文件夹的注释。我如何可视化数据集 XML 像这样的注释

以及如何将其用于 YOLOv5 自定义训练?我已阅读 this 教程,但此数据集每个文件夹(包含多个图像)只有 1 XML 个注释,而不是 1 个图像的 1 XML。

您可以使用 labelimg 可视化所有符号和图像 https://github.com/tzutalin/labelImg

要将xml(pascal/voc格式)转换成txt(yolo格式)可以使用labelimg程序,然后点击“pascal/voc”,会变成“yolo 格式”。单击支票图像并保存。该程序将以 yolo 格式保存您所在的图像。

对于自动转换,您可以使用此示例作为模板来编写您自己的转换器https://github.com/rafael-junio/mio-dataset-converter/blob/master/mio-dataset-converter.py