Spaces:
Running on Zero
Running on Zero
| import supervision as sv | |
| from PIL import Image | |
| def annotate_with_boxes(image: Image, detections: sv.Detections) -> Image: | |
| annotated_image = image.copy() | |
| thickness = sv.calculate_optimal_line_thickness(resolution_wh=image.size) | |
| text_scale = sv.calculate_optimal_text_scale(resolution_wh=image.size) | |
| bounding_box_annotator = sv.BoundingBoxAnnotator( | |
| color_lookup=sv.ColorLookup.INDEX, thickness=thickness) | |
| label_annotator = sv.LabelAnnotator( | |
| color_lookup=sv.ColorLookup.INDEX, | |
| text_scale=text_scale, | |
| text_thickness=thickness) | |
| annotated_image = bounding_box_annotator.annotate(annotated_image, detections) | |
| annotated_image = label_annotator.annotate(annotated_image, detections) | |
| return annotated_image | |