Fix empty mask overlay handling when no objects are detected. (#350)

opengeos · Oct 22, 2024 · 7d1d648 · 7d1d648
1 parent 8888ba6
commit 7d1d648
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/samgeo/text_sam.py b/samgeo/text_sam.py
@@ -342,7 +342,10 @@ def predict(
 
         if boxes.nelement() == 0:  # No "object" instances found
             print("No objects found in the image.")
-            return
+            mask_overlay = np.zeros_like(
+                image_np[..., 0], dtype=dtype
+            )  # Create an empty mask overlay
+
         else:
             # Create an empty image to store the mask overlays
             mask_overlay = np.zeros_like(