cledouxluma commited on
Commit
37dd2a8
·
verified ·
1 Parent(s): 20cc740

Upload configs/scrfd_34g.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. configs/scrfd_34g.yaml +65 -0
configs/scrfd_34g.yaml ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # SCRFD-34GF — Flagship Quality Model
2
+ # WiderFace: 96.1/95.0/85.2 (Easy/Med/Hard) @ 80 FPS VGA on V100
3
+ # Params: 9.80M | GFLOPs: 34
4
+
5
+ model:
6
+ name: scrfd_34g
7
+ backbone:
8
+ stages: [3, 12, 28, 3]
9
+ widths: [56, 88, 248, 304]
10
+ block_type: basic
11
+ neck:
12
+ out_channels: 64
13
+ use_gn: true
14
+ head:
15
+ feat_channels: 64
16
+ stacked_convs: 3
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 # Generalized Focal Loss
29
+ reg: diou # Distance-IoU
30
+ cls_weight: 1.0
31
+ reg_weight: 2.0
32
+
33
+ training:
34
+ epochs: 640
35
+ batch_size: 8 # per GPU (paper: 8 × 4 GPUs)
36
+ optimizer:
37
+ type: sgd
38
+ lr: 0.01
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 # Mixed precision
48
+ grad_clip: 35.0
49
+ pretrained: null # Train from scratch (SCRFD finding)
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