Skip to content

Instantly share code, notes, and snippets.

@peune

peune/use_mod.py

Created February 16, 2018 15:47
Show Gist options
  • Save peune/db422addbdf3878a968bde5601c626ed to your computer and use it in GitHub Desktop.
Save peune/db422addbdf3878a968bde5601c626ed to your computer and use it in GitHub Desktop.
Download ZIP
Raw
use_mod.py
input_shape = (W,W,3)
base_model = create_base_model( org_model, input_shape )
style_mod = create_style_mod( base_model, input_shape )
content_mod = create_content_mod( base_model, input_shape )
mod = create_merge_model( input_shape, style_mod, content_mod )
style_target = style_mod.predict(S)
content_target = content_mod.predict(C)
mod.fit( np.ones((1,1)), [style_target, content_target], batch_size=1, epochs=1000, verbose=1 )
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment