python删除格式错误的txt文件

 

"""
有些标注的txt文件里面是错误的，例如目标后面的坐标值本来应该是pedestrian,1138,306,18,56
但是它后面的坐标只有两个，pedestrian,1138,306这样在后面进行txt to xml转换的时候会发生错误，
因此编写脚本把这种错误的txt找出来，删掉。
"""

import os 

delete_labels = []
labels_dir = "./labels"
#labels_dir = "./delete_labels"

for label in os.listdir(labels_dir):
    with open(os.path.join(labels_dir, label), 'r') as f:
        for line in f.readlines():
            if 5 != len(line.split(",")):#说明坐标是少的，这种要删除，
                print(label)
                delete_labels.append(label)

for label in delete_labels:
    os.remove(os.path.join(labels_dir, label))