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

373023820

于 2024-04-11 11:26:36 发布

阅读量113

点赞数 1

文章标签： YOLO

本文链接：https://blog.youkuaiyun.com/qq_53821866/article/details/137632519

版权

本文介绍了一个Python脚本，它从指定的labels文件夹中查找类别为1的txt文件，然后删除与这些txt文件对应的jpg图像。该脚本适用于Ultralytics项目中的YOLOv8对象检测模型。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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文件已删除。")