🚑️ [Fix] bbox bugs, choice 1 st bbox when draw

WongKinYiu · Aug 5, 2024 · dc88787 · dc88787
1 parent b8b460c
commit dc88787
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolo/tools/drawer.py b/yolo/tools/drawer.py
@@ -32,7 +32,7 @@ def draw_bboxes(
             img = img[0]
         img = to_pil_image(img)
 
-    if bboxes.ndim == 3:
+    if isinstance(bboxes, list) or bboxes.ndim == 3:
         bboxes = bboxes[0]
 
     img = img.copy()