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

Question

我正在为 YOLOv5 进行定制训练。我想使用 UA-DETRAC 中的数据集训练 YOLOv5，其中包含包含一些图像的文件夹的 XML 注释。我怎样才能像这样可视化数据集 XML 注释

以及如何使用它进行YOLOv5定制训练？我已阅读 this 教程，但此数据集每个文件夹只有 1 个 XML 注释（其中包含多个图像），而不是 1 个图像对应 1 个 XML。

Answer 1

您可以使用 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

Answer 2

嘿，你解决问题了吗？如果是的话怎么办？