Checks and separates corrupt images

check_corrupt_images.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# pip install Pillow

import os
import shutil
from PIL import Image

IMAGES_PATH = './images'
CORRUPT_IMAGES_PATH = './corrupt_images'
EXTENSIONS = ('.jpg', '.png')

os.makedirs(CORRUPT_IMAGES_PATH, exist_ok=True)

for filename in os.listdir(IMAGES_PATH):
	if filename.endswith(EXTENSIONS):

		image_path = os.path.join(IMAGES_PATH, filename)

		try:
			img = Image.open(image_path) # open the image file
			img.verify() # verify that it is, in fact an image

		except (IOError, SyntaxError) as e:
			print('Bad file:', filename) # print out the names of corrupt files
			shutil.move(image_path, os.path.join(CORRUPT_IMAGES_PATH, filename))