统计VOC中各个类别的标注数量

最新推荐文章于 2024-04-17 20:28:56 发布

「已注销」

最新推荐文章于 2024-04-17 20:28:56 发布

阅读量351

点赞数 1

本文链接：https://blog.csdn.net/zy_join/article/details/127401943

版权

XML解析 ElementTree 对象统计文件处理 Python

关键词由CSDN通过智能技术生成

import glob
import xml.etree.ElementTree as ET
#import xml.dom.minidom
import tqdm

# 类名和数量
classes_numbers = {}

#xml路径
path = r''
for xml_file in tqdm.tqdm(glob.glob(path + '/*.xml')):
    # 返回解析树
    tree = ET.parse(xml_file)
    # 获取根节点
    root = tree.getroot()
    # 对所有目标进行解析
    for member in root.findall('object'):
        #获取object标签内的name，找不到的话打开xml文件，看看是name还是clasname...
        objectname = member.find('name').text
        classes_numbers[objectname] = classes_numbers.get(objectname,0)+1
print("done")
print(classes_numbers)