使用python读写VOC2007形式存放的xml标签文件

一、前言

  本节主要记录,使用python读写VOC2007形式存放的xml标签文件

二、XML 标签文件格式

打开一个已打标注的XML文件,如下所示:

三、使用python读写其中的标注信息xmin ymin xmax ymax

import xml.etree.ElementTree as ET

def fixXml( xml_path):
    
    tree=ET.parse(xml_path)#打开文件     
    root = tree.getroot()  #

    size = root.find("size")
    width = int(size.find("width").text)  #读取宽
    height = int(size.find("height").text) #读取高
    
    for obj in root.findall('object'):
        bbox=obj.find('bndbox')
        #print(bbox)
        #获取bbox坐标信息
        x1=int(bbox.find('xmin').text)
        y1=int(bbox.find('ymin').text)
        x2=int(bbox.find('xmax').text)
        y2=int(bbox.find('ymax').text)
        
        #修改bbox坐标信息
        bbox.find('ymin').text =str(1)
        bbox.find('ymax').text =str(2)            
        bbox.find('xmin').text =str(3)
        bbox.find('xmax').text =str(4)

    root.append(size)  #可添加"size"节点        

    tree.write(xml_path)

猜你喜欢

转载自blog.csdn.net/AP1005834/article/details/109103349
今日推荐