From 3fad4786bb1bfe3ff0b2c2aa2673b85057d2358e Mon Sep 17 00:00:00 2001 From: Naveen V Date: Tue, 1 Oct 2019 19:35:51 +0530 Subject: [PATCH 1/2] Update train.py replacement of depreciated scipy.misc.imresize with PIL.Image.resize() --- train.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index 66ebf38f..ac39868d 100755 --- a/train.py +++ b/train.py @@ -5,12 +5,11 @@ import time import random import numpy as np -import scipy, multiprocessing import tensorflow as tf import tensorlayer as tl from model import get_G, get_D from config import config - +from PIL import Image ###====================== HYPER-PARAMETERS ===========================### ## Adam batch_size = config.TRAIN.batch_size # use 8 if your GPU memory is small, and change [4, 4] in tl.vis.save_images to [2, 4] @@ -184,7 +183,7 @@ def evaluate(): tl.vis.save_image(valid_lr_img[0], os.path.join(save_dir, 'valid_lr.png')) tl.vis.save_image(valid_hr_img, os.path.join(save_dir, 'valid_hr.png')) - out_bicu = scipy.misc.imresize(valid_lr_img[0], [size[0] * 4, size[1] * 4], interp='bicubic', mode=None) + out_bicu = np.array(Image.fromarray(valid_lr_img[0]).resize((size[0] * 4, size[1] * 4), resample=Image.BICUBIC)) tl.vis.save_image(out_bicu, os.path.join(save_dir, 'valid_bicubic.png')) From c4b51883708001b928b1e6f31f31d0387d59791b Mon Sep 17 00:00:00 2001 From: Naveen V Date: Tue, 1 Oct 2019 19:37:22 +0530 Subject: [PATCH 2/2] Update train.py --- train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train.py b/train.py index ac39868d..e479568e 100755 --- a/train.py +++ b/train.py @@ -10,6 +10,7 @@ from model import get_G, get_D from config import config from PIL import Image + ###====================== HYPER-PARAMETERS ===========================### ## Adam batch_size = config.TRAIN.batch_size # use 8 if your GPU memory is small, and change [4, 4] in tl.vis.save_images to [2, 4]