【Python】生成voc格式的xml标签

怼烎@ 2022-10-15 10:54 190阅读 0赞

## xml标签 ##

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1NlZV9TdGFy_size_16_color_FFFFFF_t_70]

## 代码 ##

import xml.etree.ElementTree as ET
    
    class GenAnnotations:
        def __init__(self, filename, witdh, height, depth=3, foldername="VOC2007", sourcename="Unknown"):
            self.root = ET.Element("annotation")
            self.foleder = ET.SubElement(self.root, "folder")
            self.filename = ET.SubElement(self.root, "filename")
            self.source = ET.SubElement(self.root, "source")
            self.size = ET.SubElement(self.root, "size")
            self.width = ET.SubElement(self.size, "width")
            self.height = ET.SubElement(self.size, "height")
            self.depth = ET.SubElement(self.size, "depth")
    
            self.foleder.text = foldername
            self.filename.text = filename
            self.source.text = sourcename
            self.width.text = str(witdh)
            self.height.text = str(height)
            self.depth.text = str(depth)
    
        def savefile(self, filename):
            tree = ET.ElementTree(self.root)
            tree.write(filename, xml_declaration=False, encoding='utf-8')
    
        def add_object(self, label, xmin, ymin, xmax, ymax, tpose=0, ttruncated=0, tdifficult=0):
            object = ET.SubElement(self.root, "object")
            namen = ET.SubElement(object, "name")
            namen.text = label
            pose = ET.SubElement(object, "pose")
            pose.text = str(tpose)
            truncated = ET.SubElement(object, "truncated")
            truncated.text = str(ttruncated)
            difficult = ET.SubElement(object, "difficult")
            difficult.text = str(tdifficult)
            bndbox = ET.SubElement(object, "bndbox")
            xminn = ET.SubElement(bndbox, "xmin")
            xminn.text = str(xmin)
            yminn = ET.SubElement(bndbox, "ymin")
            yminn.text = str(ymin)
            xmaxn = ET.SubElement(bndbox, "xmax")
            xmaxn.text = str(xmax)
            ymaxn = ET.SubElement(bndbox, "ymax")
            ymaxn.text = str(ymax)
    
    
    if __name__ == '__main__':
       filename="000001.jpg"
       anno= GenAnnotations(filename, 1280, 720)
       for i in range(3):
           xmin=i+1
           ymin=i+10
           xmax=i+100
           ymax=i+100
           anno.add_object("pos",xmin,ymin,xmax,ymax)
           anno.savefile("00001.xml")

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1NlZV9TdGFy_size_16_color_FFFFFF_t_70]: /images/20221014/bbc98980d1074b0884e75c5cc86c3319.png