Spaces:
Running
on
Zero
Running
on
Zero
Update utils/sam.py
Browse files- utils/sam.py +1 -0
utils/sam.py
CHANGED
|
@@ -40,6 +40,7 @@ def run_sam_inference(
|
|
| 40 |
mask, score, _ = model.predict(box=detections.xyxy, multimask_output=False)
|
| 41 |
|
| 42 |
# dirty fix; remove this later
|
|
|
|
| 43 |
if len(mask.shape) == 4:
|
| 44 |
mask = np.squeeze(mask)
|
| 45 |
|
|
|
|
| 40 |
mask, score, _ = model.predict(box=detections.xyxy, multimask_output=False)
|
| 41 |
|
| 42 |
# dirty fix; remove this later
|
| 43 |
+
print("Mask Shape lenght: ", len(mask.shape))
|
| 44 |
if len(mask.shape) == 4:
|
| 45 |
mask = np.squeeze(mask)
|
| 46 |
|