Pascal VOC格式数据集，根据指定类别名称查找包含该类别名对应的图像

大鹏要高飞

于 2023-06-27 05:44:29 发布

阅读量149

点赞数

文章标签： pascal voc

本文链接：https://blog.csdn.net/a18838956649/article/details/131408752

版权

import os
import xml.etree.ElementTree as ET

def find_xmls_with_class(xml_dir, class_name):
    xml_files_with_class = []

    # 文件名（不带扩展名）
    file_names = [name.split('.')[0] for name in os.listdir(xml_dir)]

    for file_name in file_names:
        xml_path = os.path.join(xml_dir, file_name+'.xml')

        tree = ET.parse(xml_path)
        root = tree.getroot()

        for obj in root.findall('object'):
            name = obj.find('name').text
            if name == class_name:
                xml_files_with_class.append(file_name+'.xml')
                break

    return xml_files_with_class

# 你的目录
dir_path = './'

# 指定类别名称
class_name = 'airplane'

# 获取标注文件
xml_dir = os.path.join(dir_path, 'Annotations')

xml_files_with_class = find_xmls_with_class(xml_dir, class_name)

# 输出包含该类别的xml文件
for xml_file in xml_files_with_class:
    print(xml_file)