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)