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-10 13:59:43 +01:00
parent 04f9a4d021
commit aabcb70c8c
2 changed files with 1 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
ComixGAN/model.py
View file

@ -13,7 +13,7 @@ class ComixGAN:
raise FileNotFoundError(errno.ENOENT, os.strerror(errno.ENOENT), settings.COMIX_GAN_MODEL_PATH)
self.graph = tf.Graph()
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.7
config.gpu_options.per_process_gpu_memory_fraction = 0.8
config.gpu_options.allow_growth = True
self.session = tf.Session(graph=self.graph, config=config)
with self.graph.as_default():

3
style_transfer/style_transfer.py
View file

@ -20,9 +20,6 @@ class StyleTransfer():
@classmethod
@profile
def get_stylized_frames(cls, frames, style_transfer_mode=0, gpu=settings.GPU):
print(frames[0].shape)
print(frames[0].max())
print(frames[0].min())
if style_transfer_mode == 0:
return cls._comix_gan_stylize(frames=frames)
elif style_transfer_mode == 1: