From e6b926a5e429d5a4df5edbedfed6eba9e7c2f608 Mon Sep 17 00:00:00 2001 From: awadelrahman Date: Wed, 17 Feb 2021 00:44:09 +0100 Subject: [PATCH] fix: use PIL.Image instead of scipy.imresize in util.py --- util.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/util.py b/util.py index 44b48e9..ce4c867 100644 --- a/util.py +++ b/util.py @@ -2,7 +2,7 @@ import matplotlib.pyplot as plt import numpy as np from torchvision import datasets -from scipy.misc import imresize +from PIL import Image def show_result(G, x_, y_, num_epoch, show = False, save = False, path = 'result.png'): # G.eval() @@ -81,7 +81,7 @@ def data_load(path, subfolder, transform, batch_size, shuffle=True): def imgs_resize(imgs, resize_scale = 286): outputs = torch.FloatTensor(imgs.size()[0], imgs.size()[1], resize_scale, resize_scale) for i in range(imgs.size()[0]): - img = imresize(imgs[i].numpy(), [resize_scale, resize_scale]) + img = np.array(Image.fromarray(imgs[i].numpy(), mode="RGB").resize((resize_scale, resize_scale))) outputs[i] = torch.FloatTensor((img.transpose(2, 0, 1).astype(np.float32).reshape(-1, imgs.size()[1], resize_scale, resize_scale) - 127.5) / 127.5) return outputs