From ce0e178763b94f686beb0cef9e476a213f9e0ed3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Maciej=20P=C4=99=C5=9Bko?= Date: Fri, 2 Nov 2018 22:03:41 +0100 Subject: [PATCH] Fix concatenate bug --- style_transfer/style_transfer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/style_transfer/style_transfer.py b/style_transfer/style_transfer.py index 51226db..3365996 100644 --- a/style_transfer/style_transfer.py +++ b/style_transfer/style_transfer.py @@ -60,7 +60,7 @@ class StyleTransfer(): stylized_batch_of_imgs = comixGAN_model.predict(batch_of_frames) stylized_imgs.append(255*stylized_batch_of_imgs) - return list(np.stack(stylized_imgs)) + return list(np.concatenate(stylized_imgs, axis=0)) @classmethod def _cartoon_gan_stylize(cls, frames, gpu=True, style='Hayao'):