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

mtcnn batching

Browse Source
This commit is contained in:
galthran-wq 2025-02-12 16:35:16 +00:00
parent bbf6a55037
commit ba2ff90ac4
1 changed files with 38 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
deepface/models/face_detection/MtCnn.py
View File

@ -1,5 +1,5 @@
# built-in dependencies # built-in dependencies
from typing import List from typing import List, Union
# 3rd party dependencies # 3rd party dependencies
import numpy as np import numpy as np
@ -17,44 +17,58 @@ class MtCnnClient(Detector):
def __init__(self): def __init__(self):
self.model = MTCNN() self.model = MTCNN()
def detect_faces(self, img: np.ndarray) -> List[FacialAreaRegion]: def detect_faces(
self,
img: Union[np.ndarray,
List[np.ndarray]]
) -> Union[List[FacialAreaRegion], List[List[FacialAreaRegion]]]:
""" """
Detect and align face with mtcnn Detect and align faces with mtcnn for a list of images
Args: Args:
img (np.ndarray): pre-loaded image as numpy array imgs (Union[np.ndarray, List[np.ndarray]]):
pre-loaded image as numpy array or a list of those
Returns: Returns:
results (List[FacialAreaRegion]): A list of FacialAreaRegion objects results (Union[List[FacialAreaRegion], List[List[FacialAreaRegion]]]):
A list of FacialAreaRegion objects for a single image or a list of lists of FacialAreaRegion objects for each image
""" """
if not isinstance(img, list):
img = [img]
resp = [] resp = []
# mtcnn expects RGB but OpenCV read BGR # mtcnn expects RGB but OpenCV read BGR
# img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) # img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
img_rgb = img[:, :, ::-1] img_rgb = [img[:, :, ::-1] for img in img]
detections = self.model.detect_faces(img_rgb) detections = self.model.detect_faces(img_rgb)
if detections is not None and len(detections) > 0: for image_detections in detections:
image_resp = []
if image_detections is not None and len(image_detections) > 0:
for current_detection in image_detections:
x, y, w, h = current_detection["box"]
confidence = current_detection["confidence"]
# mtcnn detector assigns left eye with respect to the observer
# but we are setting it with respect to the person itself
left_eye = current_detection["keypoints"]["right_eye"]
right_eye = current_detection["keypoints"]["left_eye"]
for current_detection in detections: facial_area = FacialAreaRegion(
x, y, w, h = current_detection["box"] x=x,
confidence = current_detection["confidence"] y=y,
# mtcnn detector assigns left eye with respect to the observer w=w,
# but we are setting it with respect to the person itself h=h,
left_eye = current_detection["keypoints"]["right_eye"] left_eye=left_eye,
right_eye = current_detection["keypoints"]["left_eye"] right_eye=right_eye,
confidence=confidence,
)
facial_area = FacialAreaRegion( image_resp.append(facial_area)
x=x,
y=y,
w=w,
h=h,
left_eye=left_eye,
right_eye=right_eye,
confidence=confidence,
)
resp.append(facial_area) resp.append(image_resp)
if len(resp) == 1:
return resp[0]
return resp return resp