| """ | |
| File: face_utils.py | |
| Author: Elena Ryumina and Dmitry Ryumin | |
| Description: This module contains utility functions related to facial landmarks and image processing. | |
| License: MIT License | |
| """ | |
| import numpy as np | |
| import math | |
| def norm_coordinates(normalized_x, normalized_y, image_width, image_height): | |
| x_px = min(math.floor(normalized_x * image_width), image_width - 1) | |
| y_px = min(math.floor(normalized_y * image_height), image_height - 1) | |
| return x_px, y_px | |
| def get_box(fl, w, h): | |
| idx_to_coors = {} | |
| for idx, landmark in enumerate(fl.landmark): | |
| landmark_px = norm_coordinates(landmark.x, landmark.y, w, h) | |
| if landmark_px: | |
| idx_to_coors[idx] = landmark_px | |
| x_min = np.min(np.asarray(list(idx_to_coors.values()))[:, 0]) | |
| y_min = np.min(np.asarray(list(idx_to_coors.values()))[:, 1]) | |
| endX = np.max(np.asarray(list(idx_to_coors.values()))[:, 0]) | |
| endY = np.max(np.asarray(list(idx_to_coors.values()))[:, 1]) | |
| (startX, startY) = (max(0, x_min), max(0, y_min)) | |
| (endX, endY) = (min(w - 1, endX), min(h - 1, endY)) | |
| return startX, startY, endX, endY | |