Spaces:
Sleeping
Sleeping
Tony Lian
commited on
Commit
•
e32648c
1
Parent(s):
d871568
Close the plot in SAM
Browse files- models/sam.py +1 -0
models/sam.py
CHANGED
@@ -105,6 +105,7 @@ def select_mask(masks, conf_scores, coarse_ious=None, rule="largest_over_conf",
|
|
105 |
plt.imshow(masks[ind])
|
106 |
plt.tight_layout()
|
107 |
plt.show()
|
|
|
108 |
|
109 |
return mask, selection_conf
|
110 |
|
|
|
105 |
plt.imshow(masks[ind])
|
106 |
plt.tight_layout()
|
107 |
plt.show()
|
108 |
+
plt.close()
|
109 |
|
110 |
return mask, selection_conf
|
111 |
|