python查询xml类别

最新推荐文章于 2024-11-02 16:28:26 发布

李烁.

最新推荐文章于 2024-11-02 16:28:26 发布

阅读量531

点赞数 2

分类专栏： YOLO 文章标签： python 笔记人工智能

本文链接：https://blog.csdn.net/new___dream/article/details/135962375

版权

YOLO 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

第一章导包

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)