Spaces:

sumit-ai-ml
/

SAM-with-YOLO

Runtime error

sumit-ai-ml commited on Oct 3, 2023

Commit

37e59b4

1 Parent(s): 438b834

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -62,7 +62,7 @@ def segment_image(input_image):
     fmask = masks_[0].astype(int)
-    resized_mask1 =cv2.resize(fmask[0], (target_width, target_height))
     resized_mask1 = (resized_mask1 * 255).astype(np.uint8)
     overlay_image1 = img.copy()

     fmask = masks_[0].astype(int)
+    resized_mask1 =cv2.resize(fmask, (target_width, target_height))
     resized_mask1 = (resized_mask1 * 255).astype(np.uint8)
     overlay_image1 = img.copy()