images_auto_crop.py

import argparse
import os
import sys
import traceback

from PIL import Image
from random import random


class ImageCropParams:
    def __init__(self, left_px, top_px, right_px, bottom_px):
        self.crop_left_px = left_px
        self.crop_top_px = top_px
        self.crop_right_px = right_px
        self.crop_bottom_px = bottom_px


def crop_image(image_path: str, crop_params: ImageCropParams) -> bool:

    if crop_params is None:
        print(f'Error: crop_params cannot be None')
        return False

    try:

        # with Image.open(image_path) as img:
        img = Image.open(open(image_path, 'rb'))
        width, height = img.size
        # Define the box coordinates for cropping (left, upper, right, lower)
        box = (
            crop_params.crop_left_px, crop_params.crop_top_px, width - crop_params.crop_right_px,
            height- crop_params.crop_bottom_px)

        # Crop box
        img = img.crop(box)  # as a (left, upper, right, lower)-tuple.

        # Save the flipped image
        img.save(image_path)
        img.close()

    except Exception as e:
        print(f'Error processing {image_path}: {e}, {traceback.format_exc()}')


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Crop images by size from each side.')
    parser.add_argument('folder_path', type=str, help='Path to folder containing images.')
    parser.add_argument('--crop_left_px', type=int, help='Crop pixels from left.')
    parser.add_argument('--crop_top_px', type=int, help='Crop pixels from top.')
    parser.add_argument('--crop_right_px', type=int, help='Crop pixels from right.')
    parser.add_argument('--crop_bottom_px', type=int, help='Crop pixels from bottom.')

    args = parser.parse_args()

    crop_params: ImageCropParams = ImageCropParams(args.crop_left_px, args.crop_top_px, args.crop_right_px,
                                                   args.crop_bottom_px)  # left_px, top_px, right_px, bottom_px

    folder_path = args.folder_path

    if not os.path.isdir(folder_path):
        print(f'Error: {folder_path} is not a directory')
        sys.exit(1)

    count = 0
    total = len(os.listdir(folder_path))

    for file_name in os.listdir(folder_path):
        file_path = os.path.join(folder_path, file_name)
        if os.path.isfile(file_path):
            count += 1
            file_is_image = file_name.lower().endswith(('.png', '.jpg', '.jpeg'))
            if file_is_image:
                crop_image(file_path, crop_params)
                print(f'\rProcessed {count}/{total} images.', end='', flush=True)

    print('\nDone.')  # Print a message to indicate when all images have been processed.

# Run example: python images_auto_crop.py /path/to/folder --crop_left_px 20 --crop_top_px 30 --crop_right_px 20 --crop_bottom_px 30