Stable Diffusion training image cropping utility

sd_crop.py

import os
from PIL import Image

path = 'input_image_dir_path'
save_path = 'output_image_dir_path'

for filename in os.listdir(path):
    img = Image.open(path + filename)
    if img is not None:
        width, height = img.size
        if width > height:
            left = (width - height) / 2
            right = (width + height) / 2
            top = 0
            bottom = height
        else:
            left = 0
            right = width
            top = (height - width) / 2
            bottom = (height + width) / 2
        img = img.crop((left, top, right, bottom))
        img = img.resize((512, 512), Image.LANCZOS)
        img.save(save_path + filename, 'PNG')