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

Add minor fix

Browse source
This commit is contained in:
Maciej Pęśko 2018-11-01 23:22:07 +01:00
parent d94cf2eb93
commit b5c4869b1d
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
requirements.txt
View file

@ -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

2
style_transfer/style_transfer.py
View file

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