prehistoric-systems/comixify
1
0
Fork 1
mirror of https://github.com/maciej3031/comixify.git synced 2026-03-11 08:54:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix concatenate bug

Browse source
This commit is contained in:
Maciej Pęśko 2018-11-02 22:03:41 +01:00
parent 7d03d2365a
commit ce0e178763
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
style_transfer/style_transfer.py
View file

@ -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'):