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

fix linter errors

Browse Source
This commit is contained in:
Anthr@X 2023-07-24 15:54:34 +10:00
parent fc09756cfc
commit 9bf9542273
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
deepface/detectors/YunetWrapper.py
View File

@ -6,10 +6,8 @@ from deepface.commons import functions
def build_model(): def build_model():
url = ( # pylint: disable=C0301
"https://github.com/opencv/opencv_zoo/raw/main/models/" url = "https://github.com/opencv/opencv_zoo/raw/main/models/face_detection_yunet/face_detection_yunet_2023mar.onnx"
+ "face_detection_yunet/face_detection_yunet_2023mar.onnx"
)
file_name = "face_detection_yunet_2023mar.onnx" file_name = "face_detection_yunet_2023mar.onnx"
home = functions.get_deepface_home() home = functions.get_deepface_home()
if os.path.isfile(home + f"/.deepface/weights/{file_name}") is False: if os.path.isfile(home + f"/.deepface/weights/{file_name}") is False:
@ -44,9 +42,10 @@ def detect_face(detector, image, align=True, score_threshold=0.9):
if faces is None: if faces is None:
return resp return resp
for face in faces: for face in faces:
# pylint: disable=W0105
""" """
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,
whose rows are the detected face instances, columns are the location whose rows are the detected face instances, columns are the location
of a face and 5 facial landmarks. of a face and 5 facial landmarks.
The format of each row is as follows: The format of each row is as follows:
x1, y1, w, h, x_re, y_re, x_le, y_le, x_nt, y_nt, x1, y1, w, h, x_re, y_re, x_le, y_le, x_nt, y_nt,
@ -58,7 +57,7 @@ def detect_face(detector, image, align=True, score_threshold=0.9):
""" """
(x, y, w, h, x_re, y_re, x_le, y_le) = list(map(int, face[:8])) (x, y, w, h, x_re, y_re, x_le, y_le) = list(map(int, face[:8]))
# Yunet returns negative coordinates if it thinks part of # Yunet returns negative coordinates if it thinks part of
# the detected face is outside the frame. # the detected face is outside the frame.
# We set the coordinate to 0 if they are negative. # We set the coordinate to 0 if they are negative.
x = max(x, 0) x = max(x, 0)