cledouxluma commited on
Commit
74a847d
·
verified ·
1 Parent(s): ba5db3e

Upload configs/scrfd_0.5g.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. configs/scrfd_0.5g.yaml +65 -0
configs/scrfd_0.5g.yaml ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # SCRFD-0.5GF — Mobile/Edge Model
2
+ # WiderFace: 90.6/88.1/68.5 (Easy/Med/Hard) @ 1000 FPS VGA on V100
3
+ # Params: 0.57M | GFLOPs: 0.5
4
+
5
+ model:
6
+ name: scrfd_0.5g
7
+ backbone:
8
+ stages: [2, 2, 4, 2]
9
+ widths: [16, 32, 64, 128]
10
+ block_type: basic
11
+ neck:
12
+ out_channels: 16
13
+ use_gn: true
14
+ head:
15
+ feat_channels: 16
16
+ stacked_convs: 2
17
+ num_anchors: 2
18
+ use_gn: true
19
+ use_landmarks: false
20
+ anchors:
21
+ strides: [8, 16, 32]
22
+ sizes: [[16, 32], [64, 128], [256, 512]]
23
+ ratios: [1.0]
24
+ matching:
25
+ type: atss
26
+ topk: 9
27
+ losses:
28
+ cls: gfl
29
+ reg: diou
30
+ cls_weight: 1.0
31
+ reg_weight: 2.0
32
+
33
+ training:
34
+ epochs: 640
35
+ batch_size: 32
36
+ optimizer:
37
+ type: sgd
38
+ lr: 0.04
39
+ momentum: 0.9
40
+ weight_decay: 0.0005
41
+ lr_schedule:
42
+ warmup_epochs: 3
43
+ warmup_start_lr: 0.00001
44
+ type: multistep
45
+ milestones: [440, 544]
46
+ gamma: 0.1
47
+ amp: true
48
+ grad_clip: 35.0
49
+ pretrained: null
50
+
51
+ data:
52
+ root: data/wider_face
53
+ input_size: 640
54
+ crop_scales: [0.3, 0.45, 0.6, 0.8, 1.0, 1.2, 1.4, 1.6, 1.8, 2.0]
55
+ mean: [104.0, 117.0, 123.0]
56
+ flip_prob: 0.5
57
+ min_face_size: 2
58
+ num_workers: 4
59
+ robustness_augmentation: true
60
+
61
+ inference:
62
+ score_threshold: 0.3
63
+ nms_threshold: 0.4
64
+ max_detections: 750
65
+ input_size: 640