from PIL import Image
import os
import torchvision.transforms as transforms
import torchvision.utils as vutils

LOAD_PATH = "ct_synthetic_60/synth/"
SAVE_PATH = "ct_synthetic_60_large/"

def load_img(img_path, size=512):
    img = Image.open(img_path).convert('RGB')
    transform_list = [transforms.Resize((size, size)), transforms.ToTensor(),]
    transform = transforms.Compose(transform_list)
    img = transform(img)
    return img

i = 0
for img_name in os.listdir(LOAD_PATH):
    img = load_img(LOAD_PATH + img_name)
    vutils.save_image(img, fp=SAVE_PATH + img_name, nrow = 1, padding = 10)
    i += 1
    print("Done ", str(i))