Spaces:
Sleeping
Sleeping
File size: 697 Bytes
e23acaf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | from PIL import Image, ImageDraw, ImageFont
from pathlib import Path
class ImageAnnotator:
def annotate(self, image_path, labels):
img = Image.open(image_path).convert("RGB")
draw = ImageDraw.Draw(img)
width, height = img.size
y = 20
for label in labels:
draw.rectangle(
[(20, y), (400, y + 40)],
outline="red",
width=3
)
draw.text((30, y + 8), label, fill="red")
y += 55
output_path = Path("outputs") / Path(image_path).name
output_path.parent.mkdir(exist_ok=True)
img.save(output_path)
return str(output_path)
|