| 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)) | |