全部删除特定类别的txt和他对应的jpg

import os

image_folder = r'E:\ultralytics-main\jungong.v1i.yolov8\valid\images'
label_folder = r'E:\ultralytics-main\jungong.v1i.yolov8\valid\labels'
class_to_delete = '1'

# 获取labels文件夹中类别为1的txt文件列表
txt_files_to_delete = [file for file in os.listdir(label_folder) if file.endswith('.txt') and class_to_delete in open(os.path.join(label_folder, file)).read()]

# 删除对应的txt和jpg文件
for txt_file in txt_files_to_delete:
    image_file = os.path.splitext(txt_file)[0] + '.jpg'
    if image_file in os.listdir(image_folder):
        os.remove(os.path.join(label_folder, txt_file))
        os.remove(os.path.join(image_folder, image_file))

print(f"类别为{class_to_delete}的txt和对应jpg文件已删除。")