fix: table line exlusion with connected object detection

Kohulan · Nov 17, 2023 · e2f38d7 · e2f38d7
1 parent 0a833e5
commit e2f38d7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/decimer_segmentation/complete_structure.py b/decimer_segmentation/complete_structure.py
@@ -152,7 +152,8 @@ def expand_masks(
     Returns:
         np.array: Expanded masks
     """
-    labeled_array, _ = label(np.invert(image_array))
+    image_with_exclusion = np.invert(image_array) * np.invert(exclusion_mask)
+    labeled_array, _ = label(image_with_exclusion)
     mask_array = np.zeros_like(image_array)
     for seed_pixel in seed_pixels:
         x, y = seed_pixel