Python 统计VOC数据集中类的数量

Python 统计VOC数据集中类的数量

import glob
import xml.etree.ElementTree as ET
#import xml.dom.minidom
#类名
label1 = 'xxx'
label2 = 'xxx'
total_label1 = 0
total_label2 = 0
#xml路径
path = r'Z:/xxx'
for xml_file in glob.glob(path + '/*.xml'):
    # 返回解析树
    tree = ET.parse(xml_file)
    # 获取根节点
    root = tree.getroot()
    # 对所有目标进行解析
    for member in root.findall('object'):
    #获取object标签内的name
        objectname = member.find('name').text
        if objectname == label1:
            print(objectname)
            total_label1+=1
        if objectname == label2:
            print(objectname)
            total_label2+=1
print('{}的总数量是:{}'.format(label1,total_label1))
print('{}的总数量是:{}'.format(label2,total_label2))