| """ |
| 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', |
| ] |
|
|
