常使用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
标签:Xml,xml,文件,Python,text,int,bbox,find
From: https://www.cnblogs.com/sonor/p/17122441.html