| from PIL import Image, ImageDraw, ImageFont | |
| def display_bbox_and_label(image, bbox, label): | |
| # Open the image | |
| # image = Image.open(image_path) | |
| # Initialize the drawing context | |
| draw = ImageDraw.Draw(image) | |
| # Define font and font size | |
| # font = ImageFont.load_default() | |
| font = ImageFont.load_default().font_variant(size=18) | |
| # Draw bounding box on the image | |
| draw.rectangle(bbox, outline="red", width=5) | |
| # Draw label text on the image | |
| draw.text((bbox[0], bbox[1] - 15), label, fill="green", font=font) | |
| # Display the image | |
| return image | |
| # Example usage | |
| # image_path = 'example_image.jpg' # Path to your image file | |
| # bounding_box = (100, 100, 300, 300) # Define bounding box as (x_min, y_min, x_max, y_max) | |
| # label = "Object" # Define label for the bounding box | |
| # display_bbox_and_label(image_path, bounding_box, label) | |
| layout_results = layout_predictions[0] | |
| for i in range(len(layout_results.bboxes)): | |
| b_box = layout_results.bboxes[i].bbox | |
| label = layout_results.bboxes[i].label | |
| image = display_bbox_and_label(image, b_box, label) | |
| # display_bbox_and_label(image, layout_results.bboxes[3].bbox, layout_results.bboxes[3].label) | |
| # layout_predictions |