Change yolov8n to yolov8.

Lynkeus-ai · Jun 23, 2023 · 64f93d1 · 64f93d1
1 parent 9792af4
commit 64f93d1
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/README.md b/README.md
@@ -208,7 +208,7 @@ backends = [
   'mtcnn', 
   'retinaface', 
   'mediapipe',
-  'yolov8n',
+  'yolov8',
 ]
 
 #face verification

diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py
@@ -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
 

diff --git a/deepface/detectors/FaceDetector.py b/deepface/detectors/FaceDetector.py
@@ -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)