Updated script helper

titu1994 · Mar 27, 2017 · baa9313 · baa9313
1 parent 7b6a1bf
commit baa9313
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/script_helper/Script/INetwork.py b/script_helper/Script/INetwork.py
@@ -402,10 +402,10 @@ def style_loss(style, combination, mask_path=None, nb_channels=None):
     assert K.ndim(combination) == 3
 
     if mask_path is not None:
-        style_mask = load_mask(mask_path, nb_channels)
+        style_mask = K.variable(load_mask(mask_path, nb_channels))
 
-        style = style * style_mask
-        combination = combination * style_mask
+        style = style * K.stop_gradient(style_mask)
+        combination = combination * K.stop_gradient(style_mask)
 
         del style_mask
 

diff --git a/script_helper/Script/Network.py b/script_helper/Script/Network.py
@@ -399,10 +399,10 @@ def style_loss(style, combination, mask_path=None, nb_channels=None):
     assert K.ndim(combination) == 3
 
     if mask_path is not None:
-        style_mask = load_mask(mask_path, nb_channels)
+        style_mask = K.variable(load_mask(mask_path, nb_channels))
 
-        style = style * style_mask
-        combination = combination * style_mask
+        style = style * K.stop_gradient(style_mask)
+        combination = combination * K.stop_gradient(style_mask)
 
         del style_mask