Bekhouche commited on
Commit
cbe6962
·
verified ·
1 Parent(s): 302d213

Add finetune-flow-retune-b10 checkpoint

Browse files
finetune-flow-retune-b10/au_thresholds.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "thresholds": {
3
+ "AU1": 0.33999999999999997,
4
+ "AU2": 0.30000000000000004,
5
+ "AU4": 0.14,
6
+ "AU6": 0.11,
7
+ "AU7": 0.17,
8
+ "AU10": 0.4,
9
+ "AU12": 0.33,
10
+ "AU15": 0.33999999999999997,
11
+ "AU23": 0.1,
12
+ "AU24": 0.18,
13
+ "AU25": 0.31,
14
+ "AU26": 0.23
15
+ },
16
+ "per_au_f1_default": {
17
+ "AU1": 0.523,
18
+ "AU2": 0.4031469274930895,
19
+ "AU4": 0.5032679738562091,
20
+ "AU6": 0.5004824259131634,
21
+ "AU7": 0.689247703451904,
22
+ "AU10": 0.7169319947914479,
23
+ "AU12": 0.6411389936289169,
24
+ "AU15": 0.13211845102505695,
25
+ "AU23": 0.020671834625323,
26
+ "AU24": 0.08216783216783216,
27
+ "AU25": 0.8738030994054299,
28
+ "AU26": 0.3441706357423897
29
+ },
30
+ "per_au_f1_calibrated": {
31
+ "AU1": 0.5377718597425655,
32
+ "AU2": 0.4134301270417423,
33
+ "AU4": 0.5520799213887979,
34
+ "AU6": 0.5818026484408373,
35
+ "AU7": 0.7304859136727837,
36
+ "AU10": 0.7186289470595141,
37
+ "AU12": 0.6484480829259673,
38
+ "AU15": 0.1623686723973257,
39
+ "AU23": 0.184645286686103,
40
+ "AU24": 0.2226007950028393,
41
+ "AU25": 0.8806517678302163,
42
+ "AU26": 0.3960734101579172
43
+ },
44
+ "ccc_valence": 0.3870512545108795,
45
+ "ccc_arousal": 0.26323655247688293,
46
+ "p_va": 0.3251439034938812,
47
+ "p_expr": 0.2958076064576656,
48
+ "p_au_default": 0.4525123226750635,
49
+ "p_au_calibrated": 0.5024156193622175,
50
+ "p_mtl_default": 1.0734638326266104,
51
+ "p_mtl_calibrated": 1.1233671293137641
52
+ }
finetune-flow-retune-b10/config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "train_annotations": "data/training_set_annotations.txt",
3
+ "val_annotations": "data/validation_set_annotations.txt",
4
+ "image_root": "data/cropped_aligned",
5
+ "output_dir": "/data3/salah/projects/ABAW2026/code/runs/ablation_e22_affectflow_dinov3_ft_flow_beta1.0_e20",
6
+ "backbone": "dinov3_vits16",
7
+ "backbone_weights": null,
8
+ "strict_backbone": true,
9
+ "image_size": 224,
10
+ "batch_size": 32,
11
+ "epochs": 20,
12
+ "lr": 1e-05,
13
+ "weight_decay": 0.0001,
14
+ "num_workers": 4,
15
+ "freeze_backbone": false,
16
+ "deterministic_weight": 1.0,
17
+ "flow_weight": 1.0,
18
+ "va_weight": 1.0,
19
+ "expr_weight": 1.0,
20
+ "au_weight": 1.0,
21
+ "focal_gamma": 0.0,
22
+ "label_smoothing": 0.0,
23
+ "separate_proj_heads": false,
24
+ "expr_class_weighted": false,
25
+ "balanced_sampler": false,
26
+ "au_pos_weighted": false,
27
+ "flow_steps": 20,
28
+ "flow_samples": 8,
29
+ "seed": 42,
30
+ "device": "cuda",
31
+ "smoke_test": false
32
+ }
finetune-flow-retune-b10/expr_weights.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "predictions": "code/runs/ablation_e22_affectflow_dinov3_ft_flow_beta1.0_e20/validation_deterministic_exprprobs.csv",
3
+ "annotations": "data/validation_set_annotations.txt",
4
+ "weights": {
5
+ "Neutral": 0.05,
6
+ "Anger": 2.0,
7
+ "Disgust": 0.8,
8
+ "Fear": 10.0,
9
+ "Happiness": 0.3,
10
+ "Sadness": 5.0,
11
+ "Surprise": 0.2,
12
+ "Other": 0.1
13
+ },
14
+ "p_expr_default": 0.2958076064576656,
15
+ "p_expr_calibrated": 0.35045501844874727,
16
+ "per_class_f1_default": {
17
+ "Neutral": 0.3675004701899568,
18
+ "Anger": 0.21885913853317815,
19
+ "Disgust": 0.4649859943977591,
20
+ "Fear": 0.0380517503805175,
21
+ "Happiness": 0.5261498028909329,
22
+ "Sadness": 0.17085020242914978,
23
+ "Surprise": 0.15376984126984128,
24
+ "Other": 0.4262936515699893
25
+ },
26
+ "per_class_f1_calibrated": {
27
+ "Neutral": 0.3766730401529637,
28
+ "Anger": 0.21180555555555552,
29
+ "Disgust": 0.5068618481244282,
30
+ "Fear": 0.3313043478260869,
31
+ "Happiness": 0.525,
32
+ "Sadness": 0.2821450335161487,
33
+ "Surprise": 0.14505224339274736,
34
+ "Other": 0.4247980790220476
35
+ }
36
+ }
finetune-flow-retune-b10/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a2547e2cdb68ca4d8e6834659447a00f77a270c120cfc652efeeda8180dabf4
3
+ size 321540080