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 |
|