| from PIL import Image, ImageDraw, ImageFont |
| import os |
| from pathlib import Path |
|
|
|
|
| def draw(images_path, labels_path): |
| pa = Path.cwd() / 'video-test' / 'dt-tomato' |
| pal = Path(pa / labels_path) |
| dir_list = sorted(pal.glob('*')) |
| for each in dir_list: |
| draw_labels_on_image( |
| pa, images_path, pal, os.path.splitext(os.path.split(each)[1])[0] |
| ) |
|
|
|
|
| def draw_labels_on_image(path, images_path, labels_path, file_name) -> None: |
| """Draws labels on an image using the provided labels file.""" |
| image = Image.open(f'{path}/{images_path}/{file_name}.jpg') |
| width, height = image.size |
|
|
| with open(labels_path / f"{file_name}.txt") as file: |
| for line in file: |
| class_id, x_center, y_center, width_factor, height_factor = line.split() |
| x = float(x_center) * width |
| y = float(y_center) * height |
| width_factor = int(float(width_factor) * width) |
| height_factor = int(float(height_factor) * height) |
|
|
| font = ImageFont.truetype("arial.ttf", 15) |
| draw = ImageDraw.Draw(image) |
| draw.text((x, y), class_id, font=font, fill=(255, 0, 0)) |
| draw.rectangle( |
| ( |
| (x - width_factor // 2), |
| (y - height_factor // 2), |
| (x + width_factor // 2), |
| (y + height_factor // 2), |
| ), |
| outline=(255, 255, 255), |
| width=2, |
| ) |
|
|
| image.save(f'{path}/t{file_name}.jpg', quality=95) |
|
|
|
|
| if __name__ == '__main__': |
| draw('images', 'lables') |
|
|
| |
| |
|
|
| |
| |
|
|
| |
| |
|
|
