Python之xml转txt(yolo）

最新推荐文章于 2024-09-27 08:48:11 发布

小鹏聊智驾

最新推荐文章于 2024-09-27 08:48:11 发布

阅读量852

点赞数 1

分类专栏： DL工程技巧文章标签： python

本文链接：https://blog.csdn.net/chiyukunpeng/article/details/109388573

版权

Python XML处理数据转换边界框物体检测

关键词由CSDN通过智能技术生成

DL工程技巧专栏收录该内容

32 篇文章 1 订阅

订阅专栏

import os, re
import xml.etree.ElementTree as ET

class_index = {"car": 0, "truck": 1, "chemical_vehicle": 2, "bus": 3, "chemical_sign": 4}


def getbox(box, w, h):
    xmin = float(box.find("xmin").text) / w
    ymin = float(box.find("ymin").text) / h
    xmax = float(box.find("xmax").text) / w
    ymax = float(box.find("ymax").text) / h

    return ((xmin + xmax) / 2, (ymin + ymax) / 2, xmax - xmin, ymax - ymin)


def convert(inpath, outpath):
    if not os.path.exists(outpath):
        os.makedirs(outpath)
    filelist = os.listdir(inpath)
    regex = re.compile("(.+)\\.xml")
    for file in filelist:
        filename = regex.match(file)
        if filename:
            txtfile = open(outpath + "/" + filename.group(1) + ".txt", "w")
            root = ET.parse(inpath + "/" + file).getroot()
            size = root.find("size")
            width = int(size.find("width").text)
            height = int(size.find("height").text)
            for obj in root.iter("object"):
                name = obj.find("name").text
                index = class_index[name]
                box = getbox(obj.find("bndbox"), width, height)
                txtfile.write("%s %.6f %.6f %.6f %.6f\n" % (index, *box))
            txtfile.close()
        print(file, "converted")


if __name__ == '__main__':
    convert("F:/dataset/VehicleType/xml", "F:/dataset/VehicleType/label")