File size: 902 Bytes
3007f1d | 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 | """
FaceDet Models — SCRFD-family face detection architectures.
Flagship: SCRFD-34GF (96.1/95.0/85.2 Easy/Med/Hard @ 80 FPS VGA on V100)
Real-time: SCRFD-2.5GF (93.8/92.2/77.9 Easy/Med/Hard @ 400 FPS VGA on V100)
Mobile: SCRFD-0.5GF (90.6/88.1/68.5 Easy/Med/Hard @ 1000 FPS VGA on V100)
Based on: "Sample and Computation Redistribution for Efficient Face Detection"
(Guo et al., 2021, arxiv:2105.04714)
"""
from .backbone import SCRFDBackbone, build_backbone
from .neck import PAFPN, build_neck
from .head import SCRFDHead, build_head
from .detector import SCRFD, build_detector
from .anchor import AnchorGenerator, ATSSAssigner
from .losses import GFocalLoss, DIoULoss
__all__ = [
'SCRFDBackbone', 'build_backbone',
'PAFPN', 'build_neck',
'SCRFDHead', 'build_head',
'SCRFD', 'build_detector',
'AnchorGenerator', 'ATSSAssigner',
'GFocalLoss', 'DIoULoss',
]
|