Improve masking logic

jni · Mar 30, 2016 · 765dcc6 · 765dcc6
1 parent 05a228e
commit 765dcc6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gala/agglo.py b/gala/agglo.py
@@ -390,9 +390,11 @@ def __init__(self, watershed=array([], label_dtype),
         self.merge_priority_function = merge_priority_function
         self.max_merge_score = -inf
         if mask is None:
-            self.mask = np.ones(self.watershed_r.shape, dtype=bool)
+            self.mask = np.broadcast_to([True], self.watershed_r.shape)
+            self.is_masked = False
         else:
             self.mask = morpho.pad(mask, True).ravel()
+            self.is_masked = True
         self.build_graph_from_watershed()
         self.set_feature_manager(feature_manager)
         self.set_ground_truth(gt_vol)