🚑️ [Fix] bbox bugs, choice 1 st bbox when draw
Browse files- yolo/tools/drawer.py +1 -1
yolo/tools/drawer.py
CHANGED
|
@@ -32,7 +32,7 @@ def draw_bboxes(
|
|
| 32 |
img = img[0]
|
| 33 |
img = to_pil_image(img)
|
| 34 |
|
| 35 |
-
if bboxes.ndim == 3:
|
| 36 |
bboxes = bboxes[0]
|
| 37 |
|
| 38 |
img = img.copy()
|
|
|
|
| 32 |
img = img[0]
|
| 33 |
img = to_pil_image(img)
|
| 34 |
|
| 35 |
+
if isinstance(bboxes, list) or bboxes.ndim == 3:
|
| 36 |
bboxes = bboxes[0]
|
| 37 |
|
| 38 |
img = img.copy()
|