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

new function and variable names

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-12-26 13:27:28 +00:00
parent 3d914f1337
commit 0daf502e87
1 changed files with 29 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
deepface/modules/detection.py
View File

@ -255,7 +255,7 @@ def detect_faces(
) )
return [ return [
expand_and_align_face( extract_face(
facial_area=facial_area, facial_area=facial_area,
img=img, img=img,
align=align, align=align,
@ -267,7 +267,7 @@ def detect_faces(
] ]
def expand_and_align_face( def extract_face(
facial_area: FacialAreaRegion, facial_area: FacialAreaRegion,
img: np.ndarray, img: np.ndarray,
align: bool, align: bool,
@ -304,24 +304,23 @@ def expand_and_align_face(
# we were aligning the original image before, but this comes with an extra cost # we were aligning the original image before, but this comes with an extra cost
# instead we now focus on the facial area with a margin # instead we now focus on the facial area with a margin
# and align it instead of original image to decrese the cost # and align it instead of original image to decrese the cost
expanded_face, expanded_faces_width, expanded_faces_height = expand_facial_area( sub_img, relative_x, relative_y = extract_sub_image(img=img, facial_area=(x, y, w, h))
img=img, facial_area=(x, y, w, h)
) aligned_sub_img, angle = align_img_wrt_eyes(
aligned_img, angle = align_img_wrt_eyes( img=sub_img, left_eye=left_eye, right_eye=right_eye
img=expanded_face, left_eye=left_eye, right_eye=right_eye
) )
rotated_x1, rotated_y1, rotated_x2, rotated_y2 = project_facial_area( rotated_x1, rotated_y1, rotated_x2, rotated_y2 = project_facial_area(
facial_area=( facial_area=(
expanded_faces_width, relative_x,
expanded_faces_height, relative_y,
expanded_faces_width + w, relative_x + w,
expanded_faces_height + h, relative_y + h,
), ),
angle=angle, angle=angle,
size=(expanded_face.shape[0], expanded_face.shape[1]), size=(sub_img.shape[0], sub_img.shape[1]),
) )
detected_face = aligned_img[ detected_face = aligned_sub_img[
int(rotated_y1) : int(rotated_y2), int(rotated_x1) : int(rotated_x2) int(rotated_y1) : int(rotated_y2), int(rotated_x1) : int(rotated_x2)
] ]
@ -358,11 +357,12 @@ def expand_and_align_face(
) )
def expand_facial_area( def extract_sub_image(
img: np.ndarray, facial_area: Tuple[int, int, int, int] img: np.ndarray, facial_area: Tuple[int, int, int, int]
) -> Tuple[np.ndarray, int, int]: ) -> Tuple[np.ndarray, int, int]:
""" """
Expand the facial region to ensure alignment does not shift the face outside the image. Get the sub image with given facial area while expanding the facial region
to ensure alignment does not shift the face outside the image.
This function doubles the height and width of the face region, This function doubles the height and width of the face region,
and adds black pixels if necessary. and adds black pixels if necessary.
@ -372,22 +372,21 @@ def expand_facial_area(
- facial_area (tuple of int): Representing the (x, y, w, h) of the facial area. - facial_area (tuple of int): Representing the (x, y, w, h) of the facial area.
Returns: Returns:
- expanded_face (np.ndarray): expanded facial image - extracted_face (np.ndarray): expanded facial image
- expanded_faces_width (int): adjusted x-coordinates relative to the expanded region - relative_x (int): adjusted x-coordinates relative to the expanded region
- expanded_faces_height (int): adjusted y-coordinates relative to the expanded region - relative_y (int): adjusted y-coordinates relative to the expanded region
""" """
x, y, w, h = facial_area x, y, w, h = facial_area
width_border = int(0.5 * w) relative_x = int(0.5 * w)
height_border = int(0.5 * h) relative_y = int(0.5 * h)
expanded_faces_width, expanded_faces_height = width_border, height_border
# calculate expanded coordinates # calculate expanded coordinates
x1, y1 = x - width_border, y - height_border x1, y1 = x - relative_x, y - relative_y
x2, y2 = x + w + width_border, y + h + height_border x2, y2 = x + w + relative_x, y + h + relative_y
# most of the time, the expanded region fits inside the image # most of the time, the expanded region fits inside the image
if x1 >= 0 and y1 >= 0 and x2 <= img.shape[1] and y2 <= img.shape[0]: if x1 >= 0 and y1 >= 0 and x2 <= img.shape[1] and y2 <= img.shape[0]:
return img[y1:y2, x1:x2], expanded_faces_width, expanded_faces_height return img[y1:y2, x1:x2], relative_x, relative_y
# but sometimes, we need to add black pixels # but sometimes, we need to add black pixels
# ensure the coordinates are within bounds # ensure the coordinates are within bounds
@ -396,18 +395,18 @@ def expand_facial_area(
cropped_region = img[y1:y2, x1:x2] cropped_region = img[y1:y2, x1:x2]
# create a black image # create a black image
expanded_image = np.zeros( extracted_face = np.zeros(
(h + 2 * height_border, w + 2 * width_border, img.shape[2]), dtype=img.dtype (h + 2 * relative_y, w + 2 * relative_x, img.shape[2]), dtype=img.dtype
) )
# map the cropped region # map the cropped region
start_x = max(0, width_border - x) start_x = max(0, relative_x - x)
start_y = max(0, height_border - y) start_y = max(0, relative_y - y)
expanded_image[ extracted_face[
start_y : start_y + cropped_region.shape[0], start_x : start_x + cropped_region.shape[1] start_y : start_y + cropped_region.shape[0], start_x : start_x + cropped_region.shape[1]
] = cropped_region ] = cropped_region
return expanded_image, expanded_faces_width, expanded_faces_height return extracted_face, relative_x, relative_y
def align_img_wrt_eyes( def align_img_wrt_eyes(