对图像进行裁剪后，删掉没有前景目标的图像

from os.path import isdir, abspath, getsize, join
from os import listdir, system, remove
import os

 
def get_del_name(label_path, get_remove=False):
    del_names = []
    file_names = listdir(abspath(label_path))
    for file_name in file_names:
        content = join(label_path, file_name)
        if getsize(content) == 0:
            name = file_name.split(".")[0]
            del_names.append(name)
            if get_remove:
                remove(content)

    return del_names


def del_images(img_path, del_names):
    for name in del_names:
        image_name = name + ".png"
        content = join(img_path, image_name)
        if os.path.isfile(content):
            remove(content)
               

def main():
    
    label_path = "/datasets/val/label/"
    img_path = "/datasets/val/images/"
    
    del_names = get_del_name(label_path, True)
    del_images(img_path, del_names)
    
 
if __name__ == '__main__':
    main()