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

Upload configs/scrfd_10g.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. configs/scrfd_10g.yaml +65 -0
configs/scrfd_10g.yaml ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # SCRFD-10GF — Balanced Quality/Speed Model
2
+ # WiderFace: 95.2/93.9/83.1 (Easy/Med/Hard) @ 140 FPS VGA on V100
3
+ # Params: 3.86M | GFLOPs: 10
4
+
5
+ model:
6
+ name: scrfd_10g
7
+ backbone:
8
+ stages: [3, 10, 16, 3]
9
+ widths: [36, 64, 144, 224]
10
+ block_type: basic
11
+ neck:
12
+ out_channels: 56
13
+ use_gn: true
14
+ head:
15
+ feat_channels: 56
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: 12
36
+ optimizer:
37
+ type: sgd
38
+ lr: 0.015
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