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

Merge pull request #863 from bhky/master

Browse Source 
Simplify face alignment procedure
This commit is contained in:
Sefik Ilkin Serengil 2023-12-21 09:52:37 +00:00 committed by GitHub
commit bd6393daa0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
deepface/detectors/FaceDetector.py
View File

@ -1,7 +1,5 @@
import math
from PIL import Image from PIL import Image
import numpy as np import numpy as np
from deepface.commons import distance
from deepface.detectors import ( from deepface.detectors import (
OpenCvWrapper, OpenCvWrapper,
SsdWrapper, SsdWrapper,
@ -85,47 +83,19 @@ def detect_faces(face_detector, detector_backend, img, align=True):
raise ValueError("invalid detector_backend passed - " + detector_backend) raise ValueError("invalid detector_backend passed - " + detector_backend)
def get_alignment_angle_arctan2(left_eye, right_eye):
"""
The left_eye is the eye to the left of the viewer,
i.e., right eye of the person in the image.
The top-left point of the frame is (0, 0).
"""
return float(np.degrees(
np.arctan2(right_eye[1] - left_eye[1], right_eye[0] - left_eye[0])
))
def alignment_procedure(img, left_eye, right_eye): def alignment_procedure(img, left_eye, right_eye):
# this function aligns given face in img based on left and right eye coordinates angle = get_alignment_angle_arctan2(left_eye, right_eye)
img = Image.fromarray(img)
left_eye_x, left_eye_y = left_eye img = np.array(img.rotate(angle))
right_eye_x, right_eye_y = right_eye return img
# -----------------------
# find rotation direction
if left_eye_y > right_eye_y:
point_3rd = (right_eye_x, left_eye_y)
direction = -1 # rotate same direction to clock
else:
point_3rd = (left_eye_x, right_eye_y)
direction = 1 # rotate inverse direction of clock
# -----------------------
# find length of triangle edges
a = distance.findEuclideanDistance(np.array(left_eye), np.array(point_3rd))
b = distance.findEuclideanDistance(np.array(right_eye), np.array(point_3rd))
c = distance.findEuclideanDistance(np.array(right_eye), np.array(left_eye))
# -----------------------
# apply cosine rule
if b != 0 and c != 0: # this multiplication causes division by zero in cos_a calculation
cos_a = (b * b + c * c - a * a) / (2 * b * c)
angle = np.arccos(cos_a) # angle in radian
angle = (angle * 180) / math.pi # radian to degree
# -----------------------
# rotate base image
if direction == -1:
angle = 90 - angle
img = Image.fromarray(img)
img = np.array(img.rotate(direction * angle))
# -----------------------
return img # return img anyway