File size: 971 Bytes
4eb36fb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
"""
FaceDet — Production Face Detection for Video

SCRFD-family detectors with ByteTrack tracking and temporal smoothing.

Flagship:  SCRFD-34GF  (96.1/95.0/85.2 Easy/Med/Hard)  — Quality tier
Balanced:  SCRFD-10GF  (95.2/93.9/83.1 Easy/Med/Hard)  — Balanced
Real-time: SCRFD-2.5GF (93.8/92.2/77.9 Easy/Med/Hard)  — Real-time
Mobile:    SCRFD-0.5GF (90.6/88.1/68.5 Easy/Med/Hard)  — Edge/Mobile

Quick start:
    from facedet import VideoFaceDetector
    detector = VideoFaceDetector(model_path='scrfd_34g.pth', model_name='scrfd_34g')
    for result in detector.process_video('input.mp4'):
        print(result)
"""

__version__ = '1.0.0'

from .models import build_detector, SCRFD
from .engine import VideoFaceDetector, ByteTracker, TemporalSmoother
from .evaluation import WiderFaceEvaluator, SpeedBenchmark

__all__ = [
    'build_detector', 'SCRFD',
    'VideoFaceDetector', 'ByteTracker', 'TemporalSmoother',
    'WiderFaceEvaluator', 'SpeedBenchmark',
]