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

问题描述

我正在为YOlov5进行自定义培训。我想用来自UA-DETRAC的数据集训练YOlov5,该数据集包含一个包含一些图像的文件夹的XML注释。我怎样才能像这样可视化数据集XML注释

datasets visualization

以及如何将其用于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