github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 12:05:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

return empty list if detected no face.

Browse Source
This commit is contained in:
Anthr@X 2023-07-09 00:23:31 +10:00
parent a90bf8e1c0
commit 34d1f1250a
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
deepface/detectors/YunetWrapper.py
View File

@ -1,4 +1,3 @@
import numpy as np
import cv2 import cv2
import os import os
import gdown import gdown
@ -20,7 +19,7 @@ def build_model():
return face_detector return face_detector
def detect_face(detector, image, align=True, score_threshold=0.8): def detect_face(detector, image, align=True, score_threshold=0.9):
# FaceDetector.detect_faces does not support score_threshold parameter. # FaceDetector.detect_faces does not support score_threshold parameter.
# We can set it via environment variable. # We can set it via environment variable.
score_threshold = os.environ.get("yunet_score_threshold", score_threshold) score_threshold = os.environ.get("yunet_score_threshold", score_threshold)
@ -38,6 +37,8 @@ def detect_face(detector, image, align=True, score_threshold=0.8):
detector.setInputSize((width, height)) detector.setInputSize((width, height))
detector.setScoreThreshold(score_threshold) detector.setScoreThreshold(score_threshold)
_, faces = detector.detect(image) _, faces = detector.detect(image)
if faces is None:
return resp
for face in faces: for face in faces:
""" """
The detection output faces is a two-dimension array of type CV_32F, The detection output faces is a two-dimension array of type CV_32F,