fix: expansion crash when no structures have been detected #95

Kohulan · Oct 19, 2023 · 1242a67 · 1242a67
1 parent a8e8e52
commit 1242a67
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/decimer_segmentation/decimer_segmentation.py b/decimer_segmentation/decimer_segmentation.py
@@ -227,6 +227,8 @@ def get_expanded_masks(image: np.array) -> np.array:
     """
     # Structure detection with MRCNN
     masks, bboxes, _ = get_mrcnn_results(image)
+    if len(bboxes) == 0:
+        return masks
     size = determine_depiction_size_with_buffer(bboxes)
     # Mask expansion
     expanded_masks = complete_structure_mask(