Spaces:
Sleeping
Sleeping
File size: 1,095 Bytes
20bdfba |
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 |
from PIL import Image, ImageDraw, ImageFont
from physical_boxes_define import gdf
from dotenv import load_dotenv
import os
load_dotenv()
PATH_ASSETS = os.getenv('PATH_ASSETS')
# Function to draw the bounding boxes on the image
def draw_bounding_boxes(image_path, gdf):
image = Image.open(image_path+'bird.png').convert("RGB")
# Convert the image to an editable format
draw = ImageDraw.Draw(image)
# Optional: Load a font (requires a TTF file)
# try:
font = ImageFont.truetype(PATH_ASSETS + "fonts/LiberationSans-Regular.ttf",
20)
# except IOError:
# print("default")
# font = ImageFont.load_default()
# Draw each bounding box on the image
for _, row in gdf.iterrows():
xmin, ymin, xmax, ymax = row['geometry'].bounds
draw.rectangle([xmin, ymin, xmax, ymax], outline="purple", width=2)
draw.text((xmin, ymin-22), row['name'], fill="black", font=font)
image.save(image_path+'bird_boxed.png', "PNG")
if __name__ == "__main__":
draw_bounding_boxes(PATH_ASSETS + 'images/', gdf) |