ZF/uniface
1
0
Fork 0
mirror of https://github.com/yakhyo/uniface.git synced 2025-12-30 09:02:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revise model configurations in README.md

Browse Source 
Updated model names and confidence thresholds for SCRFD and YOLOv5Face in the README.
This commit is contained in:
Yakhyokhuja Valikhujaev
2025-12-08 10:07:30 +09:00
committed by GitHub
parent 637316f077
commit f4458f0550
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
README.md
View File

@@ -171,15 +171,18 @@ from uniface.detection import RetinaFace, SCRFD
from uniface.recognition import ArcFace from uniface.recognition import ArcFace
from uniface.landmark import Landmark106 from uniface.landmark import Landmark106
from uniface.constants import SCRFDWeights
# Create detector with default settings # Create detector with default settings
detector = RetinaFace() detector = RetinaFace()
# Create with custom config # Create with custom config
detector = SCRFD( detector = SCRFD(
model_name='scrfd_10g_kps', model_name=SCRFDWeights.SCRFD_10G_KPS, # SCRFDWeights.SCRFD_500M_KPS
conf_thresh=0.8, conf_thresh=0.4,
input_size=(640, 640) input_size=(640, 640)
) )
# Or with defaults settings: detector = SCRFD()
# Recognition and landmarks # Recognition and landmarks
recognizer = ArcFace() recognizer = ArcFace()
@@ -198,6 +201,7 @@ detector = RetinaFace(
conf_thresh=0.5, conf_thresh=0.5,
nms_thresh=0.4 nms_thresh=0.4
) )
# Or detector = RetinaFace()
# YOLOv5-Face detection # YOLOv5-Face detection
detector = YOLOv5Face( detector = YOLOv5Face(
@@ -205,6 +209,7 @@ detector = YOLOv5Face(
conf_thresh=0.6, conf_thresh=0.6,
nms_thresh=0.5 nms_thresh=0.5
) )
# Or detector = YOLOv5Face
# Recognition # Recognition
recognizer = ArcFace() # Uses default weights recognizer = ArcFace() # Uses default weights