Fix when no mask result found

app.py

@@ -208,6 +208,11 @@ def process_everything(automask_model, show_mask, radius_width):
         bg_image = np.asarray(bg_image)
         masks = utils.model_predict_masks_everything(automask_model, bg_image)
         im_masked = utils.show_everything(masks)
+
+        if len(im_masked) == 0:
+            st.warning("No Masks Found", icon="❗")
+            return np.asarray(bg_image)
+
         im_masked = Image.fromarray(im_masked).convert('RGBA')
         result_image = Image.alpha_composite(Image.fromarray(bg_image).convert('RGBA'),im_masked).convert("RGB")
         result_image = result_image.resize(scaled_wh)

utils/utils.py

@@ -47,12 +47,11 @@ def get_model(checkpoint='checkpoint/sam_vit_b_01ec64.pth'):
 
 
 def show_everything(sorted_anns):
+    if len(sorted_anns) == 0:
+        return np.array([])
     #sorted_anns = sorted(anns, key=(lambda x: x['stability_score']), reverse=True)
     h, w        = sorted_anns[0]['segmentation'].shape[-2:]
     #sorted_anns = sorted_anns[:int(len(sorted_anns) * stability_score/100)]
-    if sorted_anns == []:
-        st.warning("No Masks Found", icon="❗")
-        return np.zeros((h,w,4)).astype(np.uint8)
     mask = np.zeros((h,w,4))
     for ann in sorted_anns:
         m      = ann['segmentation']