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

change interface in a special case

Browse Source
This commit is contained in:
galthran-wq 2025-02-18 10:03:27 +00:00
parent 991566ffb1
commit 70b61a7f4f
5 changed files with 31 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
deepface/models/face_detection/CenterFace.py
View File

@ -44,12 +44,12 @@ class CenterFaceClient(Detector):
Returns: Returns:
results (Union[List[FacialAreaRegion], List[List[FacialAreaRegion]]]): A list or a list of lists of FacialAreaRegion objects results (Union[List[FacialAreaRegion], List[List[FacialAreaRegion]]]): A list or a list of lists of FacialAreaRegion objects
""" """
if isinstance(img, np.ndarray): if not isinstance(img, list):
return self._process_single_image(img) img = [img]
elif isinstance(img, list): results = [self._process_single_image(single_img) for single_img in img]
return [self._process_single_image(single_img) for single_img in img] if len(results) == 1:
else: return results[0]
raise ValueError("Input must be a numpy array or a list of numpy arrays.") return results
def _process_single_image(self, single_img: np.ndarray) -> List[FacialAreaRegion]: def _process_single_image(self, single_img: np.ndarray) -> List[FacialAreaRegion]:
""" """

14
deepface/models/face_detection/Dlib.py
View File

@ -57,14 +57,14 @@ class DlibClient(Detector):
Returns: Returns:
results (Union[List[FacialAreaRegion], List[List[FacialAreaRegion]]]): A list or a list of lists of FacialAreaRegion objects results (Union[List[FacialAreaRegion], List[List[FacialAreaRegion]]]): A list or a list of lists of FacialAreaRegion objects
""" """
if isinstance(img, np.ndarray): if not isinstance(img, list):
return self._detect_faces_in_single_image(img) img = [img]
elif isinstance(img, list): results = [self._process_single_image(single_img) for single_img in img]
return [self._detect_faces_in_single_image(single_img) for single_img in img] if len(results) == 1:
else: return results[0]
raise ValueError("Input must be a numpy array or a list of numpy arrays.") return results
def _detect_faces_in_single_image(self, img: np.ndarray) -> List[FacialAreaRegion]: def _process_single_image(self, img: np.ndarray) -> List[FacialAreaRegion]:
""" """
Helper function to detect faces in a single image. Helper function to detect faces in a single image.

12
deepface/models/face_detection/FastMtCnn.py
View File

@ -27,12 +27,12 @@ class FastMtCnnClient(Detector):
Returns: Returns:
results (Union[List[FacialAreaRegion], List[List[FacialAreaRegion]]]): A list or a list of lists of FacialAreaRegion objects results (Union[List[FacialAreaRegion], List[List[FacialAreaRegion]]]): A list or a list of lists of FacialAreaRegion objects
""" """
if isinstance(img, np.ndarray): if not isinstance(img, list):
return self._process_single_image(img) img = [img]
elif isinstance(img, list): results = [self._process_single_image(single_img) for single_img in img]
return [self._process_single_image(single_img) for single_img in img] if len(results) == 1:
else: return results[0]
raise ValueError("Input must be a numpy array or a list of numpy arrays.") return results
def _process_single_image(self, img: np.ndarray) -> List[FacialAreaRegion]: def _process_single_image(self, img: np.ndarray) -> List[FacialAreaRegion]:
""" """

12
deepface/models/face_detection/MediaPipe.py
View File

@ -53,12 +53,12 @@ class MediaPipeClient(Detector):
Returns: Returns:
results (Union[List[FacialAreaRegion], List[List[FacialAreaRegion]]]): A list or a list of lists of FacialAreaRegion objects results (Union[List[FacialAreaRegion], List[List[FacialAreaRegion]]]): A list or a list of lists of FacialAreaRegion objects
""" """
if isinstance(img, np.ndarray): if not isinstance(img, list):
return self._process_single_image(img) img = [img]
elif isinstance(img, list): results = [self._process_single_image(single_img) for single_img in img]
return [self._process_single_image(single_img) for single_img in img] if len(results) == 1:
else: return results[0]
raise ValueError("Input must be a numpy array or a list of numpy arrays.") return results
def _process_single_image(self, img: np.ndarray) -> List[FacialAreaRegion]: def _process_single_image(self, img: np.ndarray) -> List[FacialAreaRegion]:
""" """

12
deepface/models/face_detection/YuNet.py
View File

@ -67,12 +67,12 @@ class YuNetClient(Detector):
Returns: Returns:
results (Union[List[FacialAreaRegion], List[List[FacialAreaRegion]]]): A list or a list of lists of FacialAreaRegion objects results (Union[List[FacialAreaRegion], List[List[FacialAreaRegion]]]): A list or a list of lists of FacialAreaRegion objects
""" """
if isinstance(img, np.ndarray): if not isinstance(img, list):
return self._process_single_image(img) img = [img]
elif isinstance(img, list): results = [self._process_single_image(single_img) for single_img in img]
return [self._process_single_image(single_img) for single_img in img] if len(results) == 1:
else: return results[0]
raise ValueError("Input must be a numpy array or a list of numpy arrays.") return results
def _process_single_image(self, img: np.ndarray) -> List[FacialAreaRegion]: def _process_single_image(self, img: np.ndarray) -> List[FacialAreaRegion]:
""" """