chose_xml

import os
import shutil

# 定义函数，用于遍历文件夹并复制文件
def copy_files(src_folder, dst_folder, file_type):
    # 遍历文件夹
    for root, dirs, files in os.walk(src_folder):
        # 遍历文件
        for file in files:
            # 判断文件类型是否为xml
            if file.endswith('.xml'):
                # 获取文件路径
                file_path = os.path.join(root, file)
                # 打开文件
                with open(file_path, 'r') as f:
                    # 读取文件内容
                    content = f.read()
                    # 判断文件内容是否包含指定的类别
                    if file_type in content:
                        # 如果包含指定的类别，复制文件到目标文件夹
                        shutil.copy(file_path, dst_folder)

# 定义源文件夹路径、目标文件夹路径和文件类型
src_folder = '/home/liu_hui/YM-factory02/YM-factory02/label-xml/dp11-v4-50.75/labels/voc'
dst_folder = '/home/liu_hui/YM-factory02/YM-factory02/label-xml/dp11-v4-50.75/labels/open'
file_type = 'door_open'

# 调用函数，复制指定类别的文件到目标文件夹
copy_files(src_folder, dst_folder, file_type)