Upload models/__init__.py with huggingface_hub
Browse files- models/__init__.py +26 -0
models/__init__.py
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""
|
| 2 |
+
FaceDet Models — SCRFD-family face detection architectures.
|
| 3 |
+
|
| 4 |
+
Flagship: SCRFD-34GF (96.1/95.0/85.2 Easy/Med/Hard @ 80 FPS VGA on V100)
|
| 5 |
+
Real-time: SCRFD-2.5GF (93.8/92.2/77.9 Easy/Med/Hard @ 400 FPS VGA on V100)
|
| 6 |
+
Mobile: SCRFD-0.5GF (90.6/88.1/68.5 Easy/Med/Hard @ 1000 FPS VGA on V100)
|
| 7 |
+
|
| 8 |
+
Based on: "Sample and Computation Redistribution for Efficient Face Detection"
|
| 9 |
+
(Guo et al., 2021, arxiv:2105.04714)
|
| 10 |
+
"""
|
| 11 |
+
|
| 12 |
+
from .backbone import SCRFDBackbone, build_backbone
|
| 13 |
+
from .neck import PAFPN, build_neck
|
| 14 |
+
from .head import SCRFDHead, build_head
|
| 15 |
+
from .detector import SCRFD, build_detector
|
| 16 |
+
from .anchor import AnchorGenerator, ATSSAssigner
|
| 17 |
+
from .losses import GFocalLoss, DIoULoss
|
| 18 |
+
|
| 19 |
+
__all__ = [
|
| 20 |
+
'SCRFDBackbone', 'build_backbone',
|
| 21 |
+
'PAFPN', 'build_neck',
|
| 22 |
+
'SCRFDHead', 'build_head',
|
| 23 |
+
'SCRFD', 'build_detector',
|
| 24 |
+
'AnchorGenerator', 'ATSSAssigner',
|
| 25 |
+
'GFocalLoss', 'DIoULoss',
|
| 26 |
+
]
|