Advertisement
Guest User

Untitled

a guest
Jul 21st, 2019
88
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.58 KB | None | 0 0
  1. # total_loss
  2.  
  3. def total_loss():
  4. loss = K.variable(0.)
  5.  
  6. # contribution of content loss
  7. feature_layers_content = outputs_dict['block5_conv2']
  8. loss += content_weight * content_loss(feature_layers_content)
  9.  
  10. # contribution of style loss
  11. feature_layers_style = ['block1_conv1', 'block2_conv1',
  12. 'block3_conv1', 'block4_conv1',
  13. 'block5_conv1']
  14. loss += total_style_loss(feature_layers_style) * style_weight
  15.  
  16. # contribution of variation loss
  17.  
  18. loss += total_variation_weight * total_variation_loss(combination_image)
  19. return loss
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement