File size: 986 Bytes
b092e67 fc85e7c b092e67 0ccc47d b092e67 0ccc47d b092e67 |
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 |
from PIL import Image, ImageDraw, ImageFont
from physical_boxes_define import gdf
# 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("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('assets/images/', gdf) |