cledouxluma commited on
Commit
4eb36fb
·
verified ·
1 Parent(s): 707c26b

Upload __init__.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. __init__.py +28 -0
__init__.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ FaceDet — Production Face Detection for Video
3
+
4
+ SCRFD-family detectors with ByteTrack tracking and temporal smoothing.
5
+
6
+ Flagship: SCRFD-34GF (96.1/95.0/85.2 Easy/Med/Hard) — Quality tier
7
+ Balanced: SCRFD-10GF (95.2/93.9/83.1 Easy/Med/Hard) — Balanced
8
+ Real-time: SCRFD-2.5GF (93.8/92.2/77.9 Easy/Med/Hard) — Real-time
9
+ Mobile: SCRFD-0.5GF (90.6/88.1/68.5 Easy/Med/Hard) — Edge/Mobile
10
+
11
+ Quick start:
12
+ from facedet import VideoFaceDetector
13
+ detector = VideoFaceDetector(model_path='scrfd_34g.pth', model_name='scrfd_34g')
14
+ for result in detector.process_video('input.mp4'):
15
+ print(result)
16
+ """
17
+
18
+ __version__ = '1.0.0'
19
+
20
+ from .models import build_detector, SCRFD
21
+ from .engine import VideoFaceDetector, ByteTracker, TemporalSmoother
22
+ from .evaluation import WiderFaceEvaluator, SpeedBenchmark
23
+
24
+ __all__ = [
25
+ 'build_detector', 'SCRFD',
26
+ 'VideoFaceDetector', 'ByteTracker', 'TemporalSmoother',
27
+ 'WiderFaceEvaluator', 'SpeedBenchmark',
28
+ ]