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

Add some comments.

Browse Source
This commit is contained in:
Vincent STRAGIER 2023-06-05 12:30:57 +02:00
parent 907db28625
commit c96a2e4456
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
deepface/detectors/MediapipeWrapper.py
View File

@ -19,10 +19,11 @@ def detect_face(face_detector, img, align=True):
results = face_detector.process(img) results = face_detector.process(img)
# if no face have been detected, return an empty list # If no face has been detected, return an empty list
if results.detections is None: if results.detections is None:
return resp return resp
# Extract the bounding box, the landmarks and the confidence score
for detection in results.detections: for detection in results.detections:
(confidence,) = detection.score (confidence,) = detection.score
@ -34,6 +35,7 @@ def detect_face(face_detector, img, align=True):
y = int(bounding_box.ymin * img_height) y = int(bounding_box.ymin * img_height)
h = int(bounding_box.height * img_height) h = int(bounding_box.height * img_height)
# Extract landmarks
left_eye = (int(landmarks[0].x * img_width), int(landmarks[0].y * img_height)) left_eye = (int(landmarks[0].x * img_width), int(landmarks[0].y * img_height))
right_eye = (int(landmarks[1].x * img_width), int(landmarks[1].y * img_height)) right_eye = (int(landmarks[1].x * img_width), int(landmarks[1].y * img_height))
# nose = (int(landmarks[2].x * img_width), int(landmarks[2].y * img_height)) # nose = (int(landmarks[2].x * img_width), int(landmarks[2].y * img_height))