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

Change yolov8n to yolov8.

Browse Source
This commit is contained in:
Vincent STRAGIER 2023-06-24 00:05:48 +02:00
parent 9792af410d
commit 64f93d1fcc
3 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -208,7 +208,7 @@ backends = [
'mtcnn', 'mtcnn',
'retinaface', 'retinaface',
'mediapipe', 'mediapipe',
'yolov8n', 'yolov8',
] ]
#face verification #face verification

10
deepface/DeepFace.py
View File

@ -116,7 +116,7 @@ def verify(
This might be convenient for low resolution images. This might be convenient for low resolution images.
detector_backend (string): set face detector backend to opencv, retinaface, mtcnn, ssd, detector_backend (string): set face detector backend to opencv, retinaface, mtcnn, ssd,
dlib, mediapipe or yolov8n. dlib, mediapipe or yolov8.
align (boolean): alignment according to the eye positions. align (boolean): alignment according to the eye positions.
@ -251,7 +251,7 @@ def analyze(
resolution images. resolution images.
detector_backend (string): set face detector backend to opencv, retinaface, mtcnn, ssd, detector_backend (string): set face detector backend to opencv, retinaface, mtcnn, ssd,
dlib, mediapipe or yolov8n. dlib, mediapipe or yolov8.
align (boolean): alignment according to the eye positions. align (boolean): alignment according to the eye positions.
@ -429,7 +429,7 @@ def find(
resolution images. resolution images.
detector_backend (string): set face detector backend to opencv, retinaface, mtcnn, ssd, detector_backend (string): set face detector backend to opencv, retinaface, mtcnn, ssd,
dlib, mediapipe or yolov8n. dlib, mediapipe or yolov8.
align (boolean): alignment according to the eye positions. align (boolean): alignment according to the eye positions.
@ -641,7 +641,7 @@ def represent(
This might be convenient for low resolution images. This might be convenient for low resolution images.
detector_backend (string): set face detector backend to opencv, retinaface, mtcnn, ssd, detector_backend (string): set face detector backend to opencv, retinaface, mtcnn, ssd,
dlib, mediapipe or yolov8n. dlib, mediapipe or yolov8.
align (boolean): alignment according to the eye positions. align (boolean): alignment according to the eye positions.
@ -726,7 +726,7 @@ def stream(
model_name (string): VGG-Face, Facenet, Facenet512, OpenFace, DeepFace, DeepID, Dlib, model_name (string): VGG-Face, Facenet, Facenet512, OpenFace, DeepFace, DeepID, Dlib,
ArcFace, SFace ArcFace, SFace
detector_backend (string): opencv, retinaface, mtcnn, ssd, dlib, mediapipe or yolov8n. detector_backend (string): opencv, retinaface, mtcnn, ssd, dlib, mediapipe or yolov8.
distance_metric (string): cosine, euclidean, euclidean_l2 distance_metric (string): cosine, euclidean, euclidean_l2

4
deepface/detectors/FaceDetector.py
View File

@ -23,7 +23,7 @@ def build_model(detector_backend):
"mtcnn": MtcnnWrapper.build_model, "mtcnn": MtcnnWrapper.build_model,
"retinaface": RetinaFaceWrapper.build_model, "retinaface": RetinaFaceWrapper.build_model,
"mediapipe": MediapipeWrapper.build_model, "mediapipe": MediapipeWrapper.build_model,
"yolov8n": YoloWrapper.build_model, "yolov8": YoloWrapper.build_model,
} }
if not "face_detector_obj" in globals(): if not "face_detector_obj" in globals():
@ -66,7 +66,7 @@ def detect_faces(face_detector, detector_backend, img, align=True):
"mtcnn": MtcnnWrapper.detect_face, "mtcnn": MtcnnWrapper.detect_face,
"retinaface": RetinaFaceWrapper.detect_face, "retinaface": RetinaFaceWrapper.detect_face,
"mediapipe": MediapipeWrapper.detect_face, "mediapipe": MediapipeWrapper.detect_face,
"yolov8n": YoloWrapper.detect_face, "yolov8": YoloWrapper.detect_face,
} }
detect_face_fn = backends.get(detector_backend) detect_face_fn = backends.get(detector_backend)