Update display_BB.py

display_BB.py

@@ -1,24 +1,27 @@
-from PIL import Image, ImageDraw
+import matplotlib.pyplot as plt
+from skimage import io
+from skimage.draw import rectangle_perimeter
+from skimage.color import label2rgb
 
-def draw_bounding_boxes(image_path, boxes, labels):
-    image = Image.open(image_path)
-    draw = ImageDraw.Draw(image)
-    
+def display_bounding_boxes(image_path, boxes, labels):
+    # Load the image
+    image = io.imread(image_path)
+
+    # Convert to RGB (if grayscale)
+    if image.ndim == 2:
+        image = label2rgb(image)
+
+    # Plot the image
+    plt.imshow(image)
+
+    # Add bounding boxes and labels
     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")
-        # cv2.rectangle(image, (xmin, ymin), (xmax, ymax), color, thickness)
-        # cv2.putText(image, label, (xmin, ymin - 5), cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, thickness)
-    
-    # if save_path:
-    #     image.save(save_path)
-    
-    return image
+        ymin, xmin, ymax, xmax = box
+        rr, cc = rectangle_perimeter(start=(ymin, xmin), end=(ymax, xmax), shape=image.shape)
+        image[rr, cc] = (0, 1, 0)  # Draw green bounding boxes
+
+        # Add label text
+        plt.text(xmin, ymin - 2, label, color='r')
 
-# 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()
+    plt.axis('off')
+    plt.show()