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)