Change yolov8n to yolov8.

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

View File

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

View File

@ -116,7 +116,7 @@ def verify(
This might be convenient for low resolution images.
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.
@ -251,7 +251,7 @@ def analyze(
resolution images.
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.
@ -429,7 +429,7 @@ def find(
resolution images.
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.
@ -641,7 +641,7 @@ def represent(
This might be convenient for low resolution images.
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.
@ -726,7 +726,7 @@ def stream(
model_name (string): VGG-Face, Facenet, Facenet512, OpenFace, DeepFace, DeepID, Dlib,
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

View File

@ -23,7 +23,7 @@ def build_model(detector_backend):
"mtcnn": MtcnnWrapper.build_model,
"retinaface": RetinaFaceWrapper.build_model,
"mediapipe": MediapipeWrapper.build_model,
"yolov8n": YoloWrapper.build_model,
"yolov8": YoloWrapper.build_model,
}
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,
"retinaface": RetinaFaceWrapper.detect_face,
"mediapipe": MediapipeWrapper.detect_face,
"yolov8n": YoloWrapper.detect_face,
"yolov8": YoloWrapper.detect_face,
}
detect_face_fn = backends.get(detector_backend)