输出标注类别

import os
from xml.etree.ElementTree import ElementTree

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

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

# 输出所有类别
print(classes)

猜你喜欢

转载自blog.csdn.net/weixin_72965172/article/details/134973858
今日推荐