YOLO删除label中txt为空的，对应的图片和XML文件

大白Fao

于 2022-10-16 19:24:29 发布

阅读量962

点赞数

文章标签： xml python 开发语言

本文链接：https://blog.youkuaiyun.com/weixin_57790393/article/details/127351712

版权

该脚本用于扫描指定路径下的所有标注文件，查找其中内容为空的文件，并记录这些文件的名称。随后，根据记录的文件名删除对应的图片及XML标注文件。有效帮助维护数据集的质量。

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

import os

path=r'labels'
# print(len(os.listdir(path))) # 956
zero = open('zero.txt','w')
zero.truncate(0)
for item in os.listdir(path):
    path_item=os.path.join(path,item)
    with open(path_item,'r') as f:
        list=f.readlines()
        if len(list)<1:
            print(path_item)
            name = item[:-4] + '\n'
            zero.writelines(name)

            print('---')
zero.close()

image_ids = open('zero.txt').read().strip().split()
for image_id in image_ids:
    image_dir = "images/%s.jpg"% (image_id)
    xml_dir = "Annotations/%s.xml"% (image_id)
    if os.path.exists( image_dir):
        os.remove(image_dir)
        os.remove(xml_dir)
        print('成功删除文件:', image_dir)
    else:
        print('未找到此文件:', image_dir)
image_ids.close()