Upload __init__.py with huggingface_hub
Browse files- __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 |
+
]
|