人工智能小脚本-读取VOC数据集目录，获取标签数量

最新推荐文章于 2025-07-10 18:19:12 发布

原创最新推荐文章于 2025-07-10 18:19:12 发布 · 218 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#1024程序员节

python脚本专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一个Python程序，通过遍历train/ANNOTATIONS目录下的.xml文件，使用ElementTree库解析XML内容，统计每个对象标签（如object节点下的name属性）的出现次数。

代码如下：

import os
import xml.etree.ElementTree as ET

dataset_image_dir = 'train/ANNOTATIONS'
dataset_anno_dir = 'train/IMAGES'

# 获取标签内容和数量
tags = {}

# 遍历目录中的每个文件
for filename in os.listdir(dataset_image_dir):
    if filename.endswith('.xml'):
        # 解析XML文件
        tree = ET.parse(os.path.join(dataset_image_dir, filename))
        root = tree.getroot()

        for obj in root.findall('object'):
            name = obj.find('name').text
            if name in tags:
                tags[name] += 1
            else:
                tags[name] = 1

for tag, count in tags.items():
    print(f'{tag}: {count}')