Spaces:
Runtime error
Runtime error
darshanjani
commited on
Commit
•
4f41f90
1
Parent(s):
f4ab1d8
Update Utilities/runtime_utils.py
Browse files
Utilities/runtime_utils.py
CHANGED
@@ -32,7 +32,7 @@ def plot_bboxes(
|
|
32 |
bboxes,
|
33 |
iou_threshold=iou_thresh,
|
34 |
threshold=thresh,
|
35 |
-
|
36 |
)
|
37 |
fig = plot_image(input_img[0].permute(1, 2, 0).detach().cpu(), nms_boxes)
|
38 |
return fig, input_img
|
|
|
32 |
bboxes,
|
33 |
iou_threshold=iou_thresh,
|
34 |
threshold=thresh,
|
35 |
+
box_format="midpoint",
|
36 |
)
|
37 |
fig = plot_image(input_img[0].permute(1, 2, 0).detach().cpu(), nms_boxes)
|
38 |
return fig, input_img
|