From 3bd458963061a662787ad22448fab55d6a7eb3ea Mon Sep 17 00:00:00 2001 From: Capivariano Date: Mon, 2 Dec 2024 14:42:00 -0300 Subject: [PATCH] update: replace face detection weights with YOLOv11n-face - Updated weights path to support YOLOv11n-face - Added integration for YOLOv11n-face model for improved face detection performance - Weights hosted at: https://github.com/akanametov/yolo-face --- deepface/models/face_detection/Yolo.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/deepface/models/face_detection/Yolo.py b/deepface/models/face_detection/Yolo.py index 77dd09b..30b7af7 100644 --- a/deepface/models/face_detection/Yolo.py +++ b/deepface/models/face_detection/Yolo.py @@ -13,10 +13,10 @@ from deepface.commons.logger import Logger logger = Logger() # Model's weights paths -WEIGHT_NAME = "yolov8n-face.pt" +WEIGHT_NAME = "yolov11n-face.pt" -# Google Drive URL from repo (https://github.com/derronqi/yolov8-face) ~6MB -WEIGHT_URL = "https://drive.google.com/uc?id=1qcr9DbgsX3ryrz2uU8w4Xm3cOrRywXqb" +# Github download URL from repo (https://github.com/akanametov/yolo-face) ~5MB +WEIGHT_URL = "https://github.com/akanametov/yolo-face/releases/download/v0.0.0/yolov11n-face.pt" class YoloClient(Detector):