sujitojha commited on
Commit
34aab41
·
1 Parent(s): 26a7516

minro changes

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -168,7 +168,7 @@ def plot_image(image, boxes):
168
 
169
  buffer = BytesIO()
170
  plt.axis('off')
171
- plt.savefig(buffer,format='png')
172
  visualization = Image.open(buffer)
173
 
174
  return visualization
 
168
 
169
  buffer = BytesIO()
170
  plt.axis('off')
171
+ plt.savefig(buffer,format='png', bbox_inches='tight', pad_inches=0)
172
  visualization = Image.open(buffer)
173
 
174
  return visualization