darshanjani commited on
Commit
632d063
1 Parent(s): 4f41f90

Update Utilities/utils.py

Browse files
Files changed (1) hide show
  1. Utilities/utils.py +2 -1
Utilities/utils.py CHANGED
@@ -273,7 +273,8 @@ def plot_image(image, boxes):
273
  bbox={"color": colors[int(class_pred)], "pad": 0},
274
  )
275
 
276
- plt.show()
 
277
 
278
  def get_evaluation_bboxes(
279
  loader,
 
273
  bbox={"color": colors[int(class_pred)], "pad": 0},
274
  )
275
 
276
+ return fig
277
+ # plt.show()
278
 
279
  def get_evaluation_bboxes(
280
  loader,