Create display_BB.py

display_BB.py

@@ -0,0 +1,22 @@
+from PIL import Image, ImageDraw
+
+def draw_bounding_boxes(image_path, boxes, labels):
+    image = Image.open(image_path)
+    draw = ImageDraw.Draw(image)
+    
+    for box, label in zip(boxes, labels):
+        x_min, y_min, x_max, y_max = box
+        draw.rectangle([x_min, y_min, x_max, y_max], outline="red", width=2)
+        draw.text((x_min, y_min), label, fill="red")
+    
+    # if save_path:
+    #     image.save(save_path)
+    
+    return image
+
+# Example usage:
+image_path = "example.jpg"
+boxes = [(100, 100, 200, 200), (300, 300, 400, 400)]  # Example bounding box coordinates
+labels = ["Object 1", "Object 2"]  # Example labels
+output_image = draw_bounding_boxes(image_path, boxes, labels, save_path="output.jpg")
+output_image.show()