resize_images.py

import argparse
import os
from PIL import Image


def resize_image(image, size):
    """Resize an image to the given size."""
    return image.resize(size, Image.ANTIALIAS)


def resize_images(image_dir, output_dir, size):
    """Resize the images in 'image_dir' and save into 'output_dir'."""
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)

    images = os.listdir(image_dir)
    num_images = len(images)
    for i, image in enumerate(images):
        if image[0] != '.':
            with open(os.path.join(image_dir, image), 'r+b') as f:
                with Image.open(f) as img:
                    img = resize_image(img, size)
                    img.save(os.path.join(output_dir, image), img.format)

            if i % 100 == 0:
                print(f'[{i}/{num_images}] Resized the images and saved into "{output_dir}".')


def resize(image_dir, output_dir, dataset_type='train', year='2014', image_size=256):
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)
    # build path for input and output dataset
    dataset = dataset_type + year
    image_dir = os.path.join(image_dir, dataset)
    output_dir = os.path.join(output_dir, dataset)

    image_size = [image_size, image_size]
    resize_images(image_dir, output_dir, image_size)