[经验] 使用Python解析Xml文件常用办法(xml库)

常使用python自带xml库解析xml文件,过程如下:

1. 使用xml.etree.ElementTree库的ET.parse()函数解析XML文件。

2. 遍历XML文件中的所有< object >元素,并提取每个对象的类别和边界框信息。

3. 将类别和边界框信息分别存储在列表中,并返回。

下面是一个使用xml库加载XML格式标签文件的示例代码:

import xml.etree.ElementTree as ET

def parse_xml_annotation(xml_file):
    tree = ET.parse(xml_file)
    root = tree.getroot()
    boxes = []
    labels = []
    for obj in root.findall('object'):
        label = obj.find('name').text
        labels.append(label)
        bbox = obj.find('bndbox')
        xmin = int(bbox.find('xmin').text)
        ymin = int(bbox.find('ymin').text)
        xmax = int(bbox.find('xmax').text)
        ymax = int(bbox.find('ymax').text)
        boxes.append([xmin, ymin, xmax, ymax])
    return boxes, labels
 
posted @ 2023-02-15 12:59  倾久  阅读(331)  评论(0)    收藏  举报