tests and analysis

data/preview.ipynb

@@ -214,8 +214,8 @@
         "    label_rectangle_color='black',\n",
         "    label_text_size=14,\n",
         "    label_text_padding=5,\n",
-        "    label_rectangle_left_padding=0,\n",
-        "    label_rectangle_top_padding=0,)\n",
+        "    label_rectangle_left_margin=0,\n",
+        "    label_rectangle_top_margin=0,)\n",
         "\n",
         "def format_name(file_name: str):\n",
         "  name_, ext_ = file_name.split('.')\n",

utils/visualize_bboxes_on_image.py

@@ -6,8 +6,19 @@ import requests
 from typing import List
 from functools import cache
 
+DEFAULTS = {
+  'width': 2,
+  'bbox_color': "red",
+  'label_text_color': "black",
+  'label_rectangle_color': "red",
+  'label_text_padding': 0,
+  'label_rectangle_left_margin': 0,
+  'label_rectangle_top_margin': 0,
+  'label_text_size': 12
+}
+
 @cache
-def get_font(path_or_url: str = 'https://github.com/googlefonts/roboto/raw/main/src/hinted/Roboto-Regular.ttf', size: int = 10):
+def get_font(path_or_url: str = 'https://github.com/googlefonts/roboto/raw/main/src/hinted/Roboto-Regular.ttf', size: int = DEFAULTS['label_text_size']):
   if urlparse(path_or_url).scheme in ["http", "https"]: # Online
       return ImageFont.truetype(requests.get(path_or_url, stream=True).raw, size=size)
   else: # Local
@@ -17,15 +28,15 @@ def visualize_bboxes_on_image(
     image: Image.Image,
     bboxes: List[List[int]],
     titles: List[str] = None,
-    width = 2,
-    bbox_color="red",
-    label_text_color="black",
-    label_rectangle_color="red",
-    convert_to_x0y0x1y1 = None,
-    label_text_padding = 2,
-    label_rectangle_left_padding=10,
-    label_rectangle_top_padding=10,
-    label_text_size = 10) -> Image.Image:
+    width = DEFAULTS["width"],
+    bbox_color = DEFAULTS["bbox_color"],
+    label_text_color = DEFAULTS["label_text_color"],
+    label_rectangle_color = DEFAULTS["label_rectangle_color"],
+    label_text_padding = DEFAULTS["label_text_padding"],
+    label_rectangle_left_margin = DEFAULTS["label_rectangle_left_margin"],
+    label_rectangle_top_margin = DEFAULTS['label_rectangle_top_margin'],
+    label_text_size = DEFAULTS["label_text_size"],
+    convert_to_x0y0x1y1 = None) -> Image.Image:
   '''
   Visualize bounding boxes on an image
   Args:
@@ -42,7 +53,8 @@ def visualize_bboxes_on_image(
     label_rectangle_top_padding: Top padding of the label rectangle
     label_text_size: Font size of the label text
   Returns:
-    Image: Image with bounding boxes'''
+    Image: Image with bounding boxes
+  '''
   image = image.copy().convert("RGB")
   draw = ImageDraw.Draw(image)
   font = get_font(size = label_text_size)
@@ -51,15 +63,43 @@ def visualize_bboxes_on_image(
     x0, y0, x1, y1 = convert_to_x0y0x1y1(bbox) if convert_to_x0y0x1y1 is not None else bbox
     draw.rectangle([x0, y0, x1, y1], outline=bbox_color, width=width)
     if title is not None:
-      text_position = (x0 + label_rectangle_left_padding, y0 - label_rectangle_top_padding)
-      text_bbox_left, text_bbox_top, text_bbox_right, text_bbox_bottom = draw.textbbox(text_position, title, font=font)
-      draw.rectangle(
-        (
-          text_bbox_left - label_text_padding, 
-          text_bbox_top - label_text_padding, 
-          text_bbox_right + label_text_padding, 
-          text_bbox_bottom + label_text_padding
-        ), 
-        fill=label_rectangle_color)
-      draw.text(text_position, title, font=font, fill=label_text_color)
+      draw_text_on_image(
+        draw, 
+        [x0, y0], 
+        title, 
+        label_text_color, 
+        label_rectangle_color, 
+        label_text_padding, 
+        label_rectangle_left_margin, 
+        label_rectangle_top_margin, 
+        label_text_size, 
+        font)
+  return image
+
+def draw_text_on_image(
+    image_or_draw: Image.Image | ImageDraw.ImageDraw,
+    text_position_xy: List[int],
+    title: str,
+    label_text_color = DEFAULTS["label_text_color"],
+    label_rectangle_color = DEFAULTS["label_rectangle_color"],
+    label_text_padding = DEFAULTS["label_text_padding"],
+    label_rectangle_left_margin = DEFAULTS["label_rectangle_left_margin"],
+    label_rectangle_top_margin = DEFAULTS['label_rectangle_top_margin'],
+    label_text_size = DEFAULTS["label_text_size"],
+    font: ImageFont.FreeTypeFont = None) -> Image.Image:
+  is_image = isinstance(image_or_draw, Image.Image)
+  image = image_or_draw.copy().convert("RGB") if is_image else None
+  font = font or get_font(size = label_text_size)
+  x0, y0 = text_position_xy
+  text_position = (x0 - label_rectangle_left_margin + label_text_padding, y0 - label_rectangle_top_margin + label_text_padding)
+  draw = ImageDraw.Draw(image) if is_image else image_or_draw
+  text_bbox_left, text_bbox_top, text_bbox_right, text_bbox_bottom = draw.textbbox(text_position, title, font=font)
+  xy = [
+    text_position[0] - label_text_padding, 
+    text_position[1] - label_text_padding, 
+    text_bbox_right + label_text_padding + label_text_padding, 
+    text_bbox_bottom + label_text_padding + label_text_padding
+  ]
+  draw.rectangle(xy, fill = label_rectangle_color)
+  draw.text(text_position, title, font=font, fill=label_text_color)
   return image