clean up corrupt images

clean_up_corrupt_pics.py

import os
from PIL import Image
import pathlib
PATH_ = pathlib.Path(__file__).parent
broken_images=[]
def check_images(PATH_):   
    for pic_class in os.listdir(PATH_):
        for pic in os.listdir(f'{PATH_}/{pic_class}'):
            try:
                img = Image.open(f'{PATH_}/{pic_class}/{pic}') 
                img.verify()
            except (IOError, SyntaxError) as e:
                print('Bad file:', f'{PATH_}/{pic_class}/{pic}')
                broken_images.append(f'{PATH_}/{pic_class}/{pic}')
    return broken_images
img_to_del = check_images(f'{PATH_}/valid')
[os.remove(pic) for pic in img_to_del]