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

some linting

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

9
deepface/modules/detection.py
View File

@ -324,6 +324,9 @@ def extract_face(
int(rotated_y1) : int(rotated_y2), int(rotated_x1) : int(rotated_x2)
]
# do not spend memory for these temporary variables anymore
del aligned_sub_img, sub_img
# restore x, y, le and re before border added
x = x - width_border
y = y - height_border
@ -353,7 +356,7 @@ def extract_face(
mouth_left=mouth_left,
mouth_right=mouth_right,
),
confidence=confidence,
confidence=confidence or 0,
)
@ -411,8 +414,8 @@ def extract_sub_image(
def align_img_wrt_eyes(
img: np.ndarray,
left_eye: Union[list, tuple],
right_eye: Union[list, tuple],
left_eye: Optional[Union[list, tuple]],
right_eye: Optional[Union[list, tuple]],
) -> Tuple[np.ndarray, float]:
"""
Align a given image horizantally with respect to their left and right eye locations