python查询xml类别

第一章 导包

import os
from xml.etree.ElementTree import ElementTree

 第二章 存储类别

# 定义一个空集合用于存储类别
classes = set()

第三章 遍历所有XML文件

# 遍历指定目录下的所有XML文件
for filename in os.listdir('/home/li/PycharmProjects/Annotations'):
    if filename.endswith('.xml'):
        # 解析XML文件
        tree = ElementTree()
        tree.parse(os.path.join('/home/li/PycharmProjects/Annotations', filename))
        # 获取XML文件中的所有类别
        for obj in tree.findall('object'):
            classes.add(obj.find('name').text)

 第四章 输出所有类别

# 输出所有类别
print(classes)

全部代码如下： 

import os
from xml.etree.ElementTree import ElementTree

# 定义一个空集合用于存储类别
classes = set()

# 遍历指定目录下的所有XML文件
for filename in os.listdir('/home/li/PycharmProjects/Annotations'):
    if filename.endswith('.xml'):
        # 解析XML文件
        tree = ElementTree()
        tree.parse(os.path.join('/home/li/PycharmProjects/Annotations', filename))
        # 获取XML文件中的所有类别
        for obj in tree.findall('object'):
            classes.add(obj.find('name').text)

# 输出所有类别
print(classes)