【Pyhton】生成voc xml标签

1.voc模板标签

<annotation>
	<folder>123456</folder>
	<filename>00000.jpg</filename>
	<path>11223344.jpg</path>
	<source>
		<database>Unknown</database>
	</source>
	<size>
		<width>2560</width>
		<height>1440</height>
		<depth>3</depth>
	</size>
	<segmented>0</segmented>
</annotation>

2.python代码

# -*- coding: utf-8 -*-
import os
import xml.etree.ElementTree as ET

def create_box(root, label, box):
    x1,y1,x2,y2 = box
    # object
    object = ET.SubElement(root, "object")

    name = ET.SubElement(object, "name")
    name.text = label

    pose = ET.SubElement(object, "pose")
    pose.text = "Unspecified"

    truncated = ET.SubElement(object, "truncated")
    truncated.text = "0"

    difficult = ET.SubElement(object, "difficult")
    difficult.text = "0"

    # bndbox
    bndbox = ET.SubElement(object, "bndbox")
    xmin = ET.SubElement(bndbox, "xmin")
    xmin.text = str(x1)
    ymin = ET.SubElement(bndbox, "ymin")
    ymin.text = str(y1)
    xmax = ET.SubElement(bndbox, "xmax")
    xmax.text = str(x2)
    ymax = ET.SubElement(bndbox, "ymax")
    ymax.text = str(y2)


    return tree


if __name__ == '__main__':
    src_xml = "example2.xml"
    dst_xml = "dst.xml"
    tree = ET.parse(src_xml)
    root = tree.getroot()
    filename = root.find("filename").text
    width = root.find("size/width").text
    height = root.find("size/height").text
    print(filename)
    print(width)
    print(height)

    create_box(root, "abc", [0,0,100,100])
    create_box(root, "bcd", [100, 100, 300, 300])

    tree.write(dst_xml)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值