github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-06 11:35:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

update: replace face detection weights with YOLOv11n-face

Browse Source 
- 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
This commit is contained in:
Capivariano 2024-12-02 14:42:00 -03:00 committed by GitHub
parent e73bdb8138
commit 3bd4589630
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
deepface/models/face_detection/Yolo.py
View File

@ -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):