prepare_images.py

import PIL
from PIL import Image, ImageOps
import os

IMG_WIDTH = 256
IMG_HEIGHT = 64

def resize_pad(image):
    (img_width, img_height) = image.size
    #resize image to height 64 keeping aspect ratio
    image = image.resize((int(img_width * 64 / img_height), 64), Image.ANTIALIAS)
    (img_width, img_height) = image.size
    
    # pad image if the width is less than the max width
    if img_width > IMG_WIDTH:
        image = image.resize((IMG_WIDTH, 64), Image.ANTIALIAS)
    else:
        outImg = ImageOps.pad(image, size=(IMG_WIDTH, 64), color= "white")#, centering=(0,0)) uncommment to pad right
        image = outImg
    return image


iam_path = '/path/to/iam/words'
save_dir = '/path/to/save/processed/images'

image_paths = [os.path.join(root, file)
               for root, _, files in os.walk(iam_path)
               for file in files]

for image_path in image_paths:
    image_name = image_path.split('/')[-1]
    print(image_name)
    
    save_path = os.path.join(save_dir, image_name)
    if not os.path.exists(save_dir):
        os.makedirs(save_dir)
    try:
        image = Image.open(image_path)
        if image.mode != 'RGB': 
            image = image.convert('RGB')
        image = resize_pad(image)
        image.save(save_path)
    except PIL.UnidentifiedImageError:
        print('Error', image_path)
        continue