Tony Lian commited on
Commit
e32648c
1 Parent(s): d871568

Close the plot in SAM

Browse files
Files changed (1) hide show
  1. 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