cledouxluma commited on
Commit
3007f1d
·
verified ·
1 Parent(s): f19fabc

Upload models/__init__.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
+ ]