找出存在某一类别的label文件

import os
import shutil

# 定义目录
labels_dir = '/home/zl/yolov5-shiyan（复件）/VOCdevkit/VOCdevkit/labels/1/indicative'  # 原始标签文件夹
class1_labels_dir = '/home/zl/yolov5-shiyan（复件）/VOCdevkit/VOCdevkit/labels/1/new'  # 存放类别1标签文件的文件夹

# 确保目标文件夹存在，如果不存在则创建
os.makedirs(class1_labels_dir, exist_ok=True)

# 遍历标签文件夹
for filename in os.listdir(labels_dir):
    if filename.endswith('.txt'):  # 确保是文本文件
        filepath = os.path.join(labels_dir, filename)

        # 打开文件并检查是否包含类别1
        with open(filepath, 'r') as file:
            lines = file.readlines()
            contains_class1 = any(line.startswith('1 ') for line in lines)  # 假设类别1在每行标签的开头

        if contains_class1:
            # 如果包含类别1，移动到新的文件夹
            shutil.move(filepath, os.path.join(class1_labels_dir, filename))
        else:
            # 否则保持在原来的文件夹中
            pass


print("移动完成！")