Upload configs/scrfd_2.5g.yaml with huggingface_hub
Browse files- configs/scrfd_2.5g.yaml +65 -0
configs/scrfd_2.5g.yaml
ADDED
|
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# SCRFD-2.5GF — Real-Time Model
|
| 2 |
+
# WiderFace: 93.8/92.2/77.9 (Easy/Med/Hard) @ 400 FPS VGA on V100
|
| 3 |
+
# Params: 0.67M | GFLOPs: 2.5
|
| 4 |
+
|
| 5 |
+
model:
|
| 6 |
+
name: scrfd_2.5g
|
| 7 |
+
backbone:
|
| 8 |
+
stages: [2, 4, 4, 3]
|
| 9 |
+
widths: [24, 48, 96, 160]
|
| 10 |
+
block_type: basic
|
| 11 |
+
neck:
|
| 12 |
+
out_channels: 40
|
| 13 |
+
use_gn: true
|
| 14 |
+
head:
|
| 15 |
+
feat_channels: 40
|
| 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: 16 # Smaller model → can fit larger batch
|
| 36 |
+
optimizer:
|
| 37 |
+
type: sgd
|
| 38 |
+
lr: 0.02 # Scale with batch size
|
| 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
|