From b5c4869b1d895d65937a59d8d00801a967892c79 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Maciej=20P=C4=99=C5=9Bko?= Date: Thu, 1 Nov 2018 23:22:07 +0100 Subject: [PATCH] Add minor fix --- requirements.txt | 2 +- style_transfer/style_transfer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index a367be5..3efd366 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,7 +10,7 @@ gast==0.2.0 grpcio==1.16.0 gunicorn==19.9.0 h5py==2.8.0 -Keras==2.2.2 +Keras==2.2.4 Keras-Applications==1.0.6 Keras-Preprocessing==1.0.5 kiwisolver==1.0.1 diff --git a/style_transfer/style_transfer.py b/style_transfer/style_transfer.py index b00af5d..9b8c104 100644 --- a/style_transfer/style_transfer.py +++ b/style_transfer/style_transfer.py @@ -55,7 +55,7 @@ class StyleTransfer(): frames = cls._resize_images(frames, size=384) frames = np.stack(frames) - frames = frames / frames.max() + frames = frames / 255 stylized_imgs = comixGAN_model.generator.predict(frames) return list(255 * stylized_imgs)