github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-12 14:27:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

update alignment logic

Browse Source 
now, left eye is the one on the left with respect to the person
instead of observer
This commit is contained in:
Sefik Ilkin Serengil 2024-03-24 16:34:01 +00:00
parent 519c48e58d
commit 2b5c139670
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
deepface/modules/detection.py
View File

@ -201,8 +201,8 @@ def align_face(
Align a given image horizantally with respect to their left and right eye locations Align a given image horizantally with respect to their left and right eye locations
Args: Args:
img (np.ndarray): pre-loaded image with detected face img (np.ndarray): pre-loaded image with detected face
left_eye (list or tuple): coordinates of left eye with respect to the you left_eye (list or tuple): coordinates of left eye with respect to the person itself
right_eye(list or tuple): coordinates of right eye with respect to the you right_eye(list or tuple): coordinates of right eye with respect to the person itself
Returns: Returns:
img (np.ndarray): aligned facial image img (np.ndarray): aligned facial image
""" """
@ -214,6 +214,6 @@ def align_face(
if img.shape[0] == 0 or img.shape[1] == 0: if img.shape[0] == 0 or img.shape[1] == 0:
return img, 0 return img, 0
angle = float(np.degrees(np.arctan2(right_eye[1] - left_eye[1], right_eye[0] - left_eye[0]))) angle = float(np.degrees(np.arctan2(left_eye[1] - right_eye[1], left_eye[0] - right_eye[0])))
img = np.array(Image.fromarray(img).rotate(angle)) img = np.array(Image.fromarray(img).rotate(angle))
return img, angle return img, angle