github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 20:15: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) 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 # restore x, y, le and re before border added
x = x - width_border x = x - width_border
y = y - height_border y = y - height_border
@ -353,7 +356,7 @@ def extract_face(
mouth_left=mouth_left, mouth_left=mouth_left,
mouth_right=mouth_right, mouth_right=mouth_right,
), ),
confidence=confidence, confidence=confidence or 0,
) )
@ -411,8 +414,8 @@ def extract_sub_image(
def align_img_wrt_eyes( def align_img_wrt_eyes(
img: np.ndarray, img: np.ndarray,
left_eye: Union[list, tuple], left_eye: Optional[Union[list, tuple]],
right_eye: Union[list, tuple], right_eye: Optional[Union[list, tuple]],
) -> Tuple[np.ndarray, float]: ) -> Tuple[np.ndarray, float]:
""" """
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