Auto-creating default semantic map for style if non-existant. Closes …

…#42.
Andrei-Colesnic · Mar 13, 2016 · a660f1f · a660f1f
1 parent 098b850
commit a660f1f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/doodle.py b/doodle.py
@@ -233,6 +233,10 @@ def __init__(self):
             self.content_map_original = np.zeros(self.content_img_original.shape[:2]+(1,))
             self.semantic_weight = 0.0
 
+        if self.style_map_original is None:
+            self.style_map_original = np.zeros(self.style_img_original.shape[:2]+(1,))
+            self.semantic_weight = 0.0
+
         if self.content_img_original is None:
             self.content_img_original = np.zeros(self.content_map_original.shape[:2]+(3,))
             args.content_weight = 0.0