🔥 [Remove] scale function in drawer

yolo/tools/drawer.py

@@ -13,7 +13,6 @@ def draw_bboxes(
     img: Union[Image.Image, torch.Tensor],
     bboxes: List[List[Union[int, float]]],
     *,
-    scaled_bbox: bool = True,
     save_path: str = "",
     save_name: str = "visualize.png",
     idx2label: Optional[list],
@@ -29,13 +28,12 @@ def draw_bboxes(
     # Convert tensor image to PIL Image if necessary
     if isinstance(img, torch.Tensor):
         if img.dim() > 3:
-            logger.warning("🔍 Multi-frame tensor detected, using the first image.")
-            img = img[0]
-            bboxes = bboxes[0]
+            logger.warning("🔍 >3 dimension tensor detected, using the 0-idx image.")
+            img, bboxes = img[0], bboxes[0]
         img = to_pil_image(img)
 
     draw = ImageDraw.Draw(img, "RGBA")
-    width, height = img.size
+
     try:
         font = ImageFont.truetype("arial.ttf", 15)
     except IOError:
@@ -43,19 +41,16 @@ def draw_bboxes(
 
     for bbox in bboxes:
         class_id, x_min, y_min, x_max, y_max, *conf = [float(val) for val in bbox]
-        if scaled_bbox:
-            x_min = x_min * width
-            x_max = x_max * width
-            y_min = y_min * height
-            y_max = y_max * height
-        shape = [(x_min, y_min), (x_max, y_max)]
+        bbox = [(x_min, y_min), (x_max, y_max)]
+
         random.seed(int(class_id))
         color_map = (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))
-        draw.rounded_rectangle(shape, outline=(*color_map, 170), radius=5)
-        draw.rounded_rectangle(shape, fill=(*color_map, 50), radius=5)
 
-        text_class = str(idx2label[int(class_id)] if idx2label else class_id)
-        label_text = f"{text_class}" + (f" {conf[0]: .0%}" if conf else "")
+        draw.rounded_rectangle(bbox, outline=(*color_map, 170), radius=5)
+        draw.rounded_rectangle(bbox, fill=(*color_map, 50), radius=5)
+
+        class_text = str(idx2label[int(class_id)] if idx2label else class_id)
+        label_text = f"{class_text}" + (f" {conf[0]: .0%}" if conf else "")
 
         text_bbox = font.getbbox(label_text)
         text_width = text_bbox[2] - text_bbox[0]
@@ -65,6 +60,7 @@ def draw_bboxes(
         draw.rounded_rectangle(text_background, fill=(*color_map, 175), radius=2)
         draw.text((x_min, y_min), label_text, fill="white", font=font)
 
+    os.makedirs(save_path, exist_ok=True)
     save_image_path = os.path.join(save_path, save_name)
     img.save(save_image_path)  # Save the image with annotations
     logger.info(f"💾 Saved visualize image at {save_image_path}")

yolo/tools/solver.py

@@ -117,7 +117,6 @@ class ModelTester:
                 draw_bboxes(
                     images[0],
                     nms_out[0],
-                    scaled_bbox=False,
                     save_path=self.save_path,
                     save_name=f"frame{idx:03d}.png",
                     idx2label=self.idx2label,