(目标检测)生成XML文件脚本——python

(目标检测)生成XML文件脚本——python

from xml.dom import minidom

def gen_xml_rect(xml_path, img_full_path, obj_list,width, height):
    """
    生成矩形框标注文件
    :param xml_path: 标注文件路径
    :param img_full_path: 图像文件路径
    :param obj_list: 目标位置信息列表  (xmin,ymin,xmax,ymax,label)
    :param width:图像宽
    :param height:图像宽
    """
    # 1.创建DOM树对象
    dom = minidom.Document()
    # 2.创建根节点。每次都要用DOM对象来创建任何节点。
    root_node = dom.createElement('annotation')
    # 3.用DOM对象添加根节点
    dom.appendChild(root_node)

    # 用DOM对象创建元素子节点
    folder_node = dom.createElement('folder')
    # 用父节点对象添加元素子节点
    root_node.appendChild(folder_node)
    img_folder_path=os.path.basename(img_full_path)
    folder__node_text = dom.createTextNode(img_folder_path)
    # 用添加了文本的节点对象(看成文本节点的父节点)添加文本节点
    folder_node.appendChild(folder__node_text)

    filename_node = dom.createElement('filename')
    root_node.appendChild(filename_node)
    file_name=img_full_path.split('\\')[-1]
    filename_node_text = dom.createTextNode(file_name)
    filename_node.appendChild(filename_node_text)

    path_node = dom.createElement('path')
    root_node.appendChild(path_node)
    path_node_text = dom.createTextNode(img_full_path)
    path_node.appendChild(path_node_text)

    source_node = dom.createElement('source')
    root_node.appendChild(source_node)

    database_node = dom.createElement('database')
    source_node.appendChild(database_node)
    database_node_text = dom.createTextNode('Unknown')
    database_node.appendChild(database_node_text)

    size_node = dom.createElement('size')
    root_node.appendChild(size_node)

    width_node = dom.createElement('width')
    size_node.appendChild(width_node)
    width_node_text = dom.createTextNode(str(width))
    width_node.appendChild(width_node_text)

    height_node = dom.createElement('height')
    size_node.appendChild(height_node)
    height_node_text = dom.createTextNode(str(height))
    height_node.appendChild(height_node_text)

    depth_node = dom.createElement('depth')
    size_node.appendChild(depth_node)
    depth_node_node_text = dom.createTextNode(str(3))
    depth_node.appendChild(depth_node_node_text)

    segmented_node = dom.createElement('segmented')
    root_node.appendChild(segmented_node)
    segmented_node_text = dom.createTextNode(str(0))
    segmented_node.appendChild(segmented_node_text)

    for obj in obj_list:
        object_node = dom.createElement('object')
        root_node.appendChild(object_node)

        name_node = dom.createElement('name')
        object_node.appendChild(name_node)
        name_node_text = dom.createTextNode(str(obj[4]))
        name_node.appendChild(name_node_text)

        pose_node = dom.createElement('pose')
        object_node.appendChild(pose_node)
        pose_node_text = dom.createTextNode('Unspecified')
        pose_node.appendChild(pose_node_text)

        truncated_node = dom.createElement('truncated')
        object_node.appendChild(truncated_node)
        truncated_node_text = dom.createTextNode(str(0))
        truncated_node.appendChild(truncated_node_text)

        difficult_node = dom.createElement('difficult')
        object_node.appendChild(difficult_node)
        difficult_node_text = dom.createTextNode('0')
        difficult_node.appendChild(difficult_node_text)

        bndbox_node = dom.createElement('bndbox')
        object_node.appendChild(bndbox_node)

        xmin_node = dom.createElement('xmin')
        bndbox_node.appendChild(xmin_node)
        xmin_node_text = dom.createTextNode(str(obj[0]))
        xmin_node.appendChild(xmin_node_text)

        ymin_node = dom.createElement('ymin')
        bndbox_node.appendChild(ymin_node)
        ymin_node_text = dom.createTextNode(str(obj[1]))
        ymin_node.appendChild(ymin_node_text)

        xmax_node = dom.createElement('xmax')
        bndbox_node.appendChild(xmax_node)
        xmax_node_text = dom.createTextNode(str(obj[2]))
        xmax_node.appendChild(xmax_node_text)

        ymax_node = dom.createElement('ymax')
        bndbox_node.appendChild(ymax_node)
        ymax_node_text= dom.createTextNode(str(obj[3]))
        ymax_node.appendChild(ymax_node_text)

    try:
        with open(xml_path, 'w', encoding='UTF-8') as fh:
            # 4.writexml()第一个参数是目标文件对象,第二个参数是根节点的缩进格式,第三个参数是其他子节点的缩进格式,
            # 第四个参数制定了换行格式,第五个参数制定了xml内容的编码。
            dom.writexml(fh, addindent=" ", newl="\n",encoding='UTF-8')
            # print('写入xml OK!')
    except Exception as err:
        print('错误信息:{0}'.format(err))


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Fire丶Chicken

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值