computer-vision - 如何将 YOLO 注释 (.txt) 转换为 PASCAL VOC (.xml)？

Question

我已经建立了一个数据集来训练 YOLOv4，并且我拥有 YOLO 格式的所有标签（我使用了 LabelImg）。现在我想用相同的数据集训练 SSD，因此我需要 PASCAL VOC 格式的标签。我已经看到了一些进行相反转换（voc 到 yolo）的方法，但不是我正在寻找的方法。由于我有成千上万的图像，我想找到一种方法来自动化整个过程，而不必逐个浏览每个图像/标签。

有没有人对如何解决这个问题有任何想法？

提前致谢！

标签： computer-visionlabelobject-detectionyolo

Answer 1

我曾经也有过一样的问题。我写了一个脚本来为你做这个转换。您可以在此处找到代码的链接：https ://github.com/JPM-Tech/Object-Detection/tree/main/Scripts/converters/convert-yolo-to-xml.py 。这是一些关于如何使用它的文档的链接：https ://github.com/JPM-Tech/Object-Detection/tree/main/Scripts/converters#convert-yolo-to-xml 。希望这对你有帮助，就像它对我有帮助一样。