Spaces:
Running
Running
Commit
·
451a26f
1
Parent(s):
65390f3
fix bug: select the mask with highest score
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ def infer(img):
|
|
40 |
point_coords=input_point,
|
41 |
point_labels=input_label,
|
42 |
)
|
43 |
-
result_label = [(masks[
|
44 |
return image, result_label
|
45 |
|
46 |
|
|
|
40 |
point_coords=input_point,
|
41 |
point_labels=input_label,
|
42 |
)
|
43 |
+
result_label = [(masks[scores.argmax(), :, :], "mask")]
|
44 |
return image, result_label
|
45 |
|
46 |
|