image resize & random crop , crop --> resize, SR train data

data_resize.py

import random
import os
from tqdm import tqdm
from PIL import Image, ImageOps
from concurrent.futures import ThreadPoolExecutor

original_path = './original/'
resize_path = './resize/'
data_path = './data/'
pbar = None

def split_name(name):
    return name.split('/')[-1].split('.')[0]

def resize_image(image_path):
    img = Image.open(image_path)
    temp_name = split_name(image_path)

    width, height = img.size

    for i in range(16):
        t_w = random.randrange(0, width - 384)
        t_h = random.randrange(0, height - 384)

        fit_img = img.crop((t_w, t_h, t_w + 384, t_h + 384))
        fit_img.save('{}{}_{}.jpg'.format(original_path, temp_name, i))

        fit_img = ImageOps.fit(fit_img, (96, 96), Image.BICUBIC)
        fit_img.save('{}{}_{}.jpg'.format(resize_path, temp_name, i))

    pbar.update(1)


if __name__ == "__main__":
    temp_arr = [os.path.join(data_path, x) for x in os.listdir(data_path)]
    print('data length: ', len(temp_arr))
    pbar = tqdm(total=len(temp_arr))
    
    with ThreadPoolExecutor() as executor:
        executor.map(resize_image, temp_arr)