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

Add flag to normalize output face image

Browse Source
This commit is contained in:
Programador Artificial 2024-07-15 10:19:26 -03:00
parent b75e37c3f6
commit bb8096ad97
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
deepface/DeepFace.py
View File

@ -483,6 +483,7 @@ def extract_faces(
align: bool = True, align: bool = True,
expand_percentage: int = 0, expand_percentage: int = 0,
grayscale: bool = False, grayscale: bool = False,
normalize_face: bool = True,
anti_spoofing: bool = False, anti_spoofing: bool = False,
) -> List[Dict[str, Any]]: ) -> List[Dict[str, Any]]:
""" """
@ -506,6 +507,9 @@ def extract_faces(
grayscale (boolean): Flag to convert the output face image to grayscale grayscale (boolean): Flag to convert the output face image to grayscale
(default is False). (default is False).
normalize_face (boolean): Flag to enable normalization (divide by 255) of the output
face image output face image normalization (default is True).
anti_spoofing (boolean): Flag to enable anti spoofing (default is False). anti_spoofing (boolean): Flag to enable anti spoofing (default is False).
Returns: Returns:
@ -535,6 +539,7 @@ def extract_faces(
align=align, align=align,
expand_percentage=expand_percentage, expand_percentage=expand_percentage,
grayscale=grayscale, grayscale=grayscale,
normalize_face=normalize_face,
anti_spoofing=anti_spoofing, anti_spoofing=anti_spoofing,
) )

9
deepface/modules/detection.py
View File

@ -25,6 +25,7 @@ def extract_faces(
align: bool = True, align: bool = True,
expand_percentage: int = 0, expand_percentage: int = 0,
grayscale: bool = False, grayscale: bool = False,
normalize_face: bool = True,
anti_spoofing: bool = False, anti_spoofing: bool = False,
) -> List[Dict[str, Any]]: ) -> List[Dict[str, Any]]:
""" """
@ -43,11 +44,14 @@ def extract_faces(
align (bool): Flag to enable face alignment (default is True). align (bool): Flag to enable face alignment (default is True).
expand_percentage (int): expand detected facial area with a percentage expand_percentage (int): expand detected facial area with a percentage.
grayscale (boolean): Flag to convert the output face image to grayscale grayscale (boolean): Flag to convert the output face image to grayscale
(default is False). (default is False).
normalize_face (boolean): Flag to enable normalization (divide by 255) of the output
face image output face image normalization (default is True).
anti_spoofing (boolean): Flag to enable anti spoofing (default is False). anti_spoofing (boolean): Flag to enable anti spoofing (default is False).
Returns: Returns:
@ -117,7 +121,8 @@ def extract_faces(
if grayscale is True: if grayscale is True:
current_img = cv2.cvtColor(current_img, cv2.COLOR_BGR2GRAY) current_img = cv2.cvtColor(current_img, cv2.COLOR_BGR2GRAY)
current_img = current_img / 255 # normalize input in [0, 1] if normalize_face:
current_img = current_img / 255 # normalize input in [0, 1]
x = int(current_region.x) x = int(current_region.x)
y = int(current_region.y) y = int(current_region.y)