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 tmp solution for GPU ram problem

Browse source
This commit is contained in:
Maciej Pęśko 2018-11-04 20:10:49 +01:00
parent bbaac2c041
commit acb0cccfb1
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

@ -57,7 +57,7 @@ class StyleTransfer():
comixGAN_model = load_model(settings.COMIX_GAN_MODEL_PATH,
custom_objects={'InstanceNormalization': InstanceNormalization})
frames = cls._resize_images(frames, size=450)
batch_size = 5
batch_size = 2
stylized_imgs = []
for i in range(0, len(frames), batch_size):
batch_of_frames = np.stack(frames[i:i + batch_size]) / 255