Fix exclusion contour strategy in boolean mask

notmatthancock · Aug 2, 2018 · 133e0ec · 133e0ec
1 parent 0654e9f
commit 133e0ec
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pylidc/Annotation.py b/pylidc/Annotation.py
@@ -1061,6 +1061,11 @@ def boolean_mask(self, pad=None, bbox=None, include_contour_points=False):
                 not_contains_pts = not_contains_pts.reshape(mask.shape[:2])
                 mask[:,:,zi] = np.logical_and(mask[:,:,zi], not_contains_pts)
 
+                # Remove the contour points themselves.
+                i, j = (C - bb[:2,0]).T
+                k = np.ones(C.shape[0], dtype=np.int)*zi
+                mask[i,j,k] = False
+
         return mask
 
     def _as_set(self):