Spaces:

darshanjani
/

YoloV3-from-Scratch

Runtime error

darshanjani commited on Sep 9, 2023

Commit

632d063

1 Parent(s): 4f41f90

Update Utilities/utils.py

Files changed (1) hide show

Utilities/utils.py CHANGED Viewed

@@ -273,7 +273,8 @@ def plot_image(image, boxes):
             bbox={"color": colors[int(class_pred)], "pad": 0},
         )
-    plt.show()
 def get_evaluation_bboxes(
     loader,

             bbox={"color": colors[int(class_pred)], "pad": 0},
         )
+    return fig
+    # plt.show()
 def get_evaluation_bboxes(
     loader,