File size: 638 Bytes
a495849 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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))
|