File size: 4,029 Bytes
6f4e394
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
[
    {
        "image_backbone": "efficientnet_b3",
        "audio_backbone": "transformer",
        "validation_mse": 0.21577325425086877,
        "validation_mae": 0.36228722945237773,
        "test_mse": 0.21746371760964395,
        "test_mae": 0.36353210285305976,
        "model_path": "test_models/efficientnet_b3_transformer_model.pt"
    },
    {
        "image_backbone": "efficientnet_b0",
        "audio_backbone": "transformer",
        "validation_mse": 0.24033201676912797,
        "validation_mae": 0.42209602166444826,
        "test_mse": 0.19470563121140003,
        "test_mae": 0.37649240642786025,
        "model_path": "test_models/efficientnet_b0_transformer_model.pt"
    },
    {
        "image_backbone": "resnet50",
        "audio_backbone": "transformer",
        "validation_mse": 0.22672857019381645,
        "validation_mae": 0.3926378931754675,
        "test_mse": 0.22427306957542897,
        "test_mae": 0.39585837423801423,
        "model_path": "test_models/resnet50_transformer_model.pt"
    },
    {
        "image_backbone": "resnet50",
        "audio_backbone": "bidirectional_lstm",
        "validation_mse": 0.2967155438203078,
        "validation_mae": 0.3850937023376807,
        "test_mse": 0.36476454623043536,
        "test_mae": 0.425818096101284,
        "model_path": "test_models/resnet50_bidirectional_lstm_model.pt"
    },
    {
        "image_backbone": "efficientnet_b0",
        "audio_backbone": "bidirectional_lstm",
        "validation_mse": 0.5120524473679371,
        "validation_mae": 0.5665570046657171,
        "test_mse": 0.5059382550418376,
        "test_mae": 0.555050653219223,
        "model_path": "test_models/efficientnet_b0_bidirectional_lstm_model.pt"
    },
    {
        "image_backbone": "efficientnet_b3",
        "audio_backbone": "bidirectional_lstm",
        "validation_mse": 0.8020018790012751,
        "validation_mae": 0.7953977386156718,
        "test_mse": 0.7042828559875488,
        "test_mae": 0.7441241115331649,
        "model_path": "test_models/efficientnet_b3_bidirectional_lstm_model.pt"
    },
    {
        "image_backbone": "efficientnet_b0",
        "audio_backbone": "gru",
        "validation_mse": 1.1340507984161377,
        "validation_mae": 0.8290961503982544,
        "test_mse": 0.9705999374389649,
        "test_mae": 0.7704607486724854,
        "model_path": "test_models/efficientnet_b0_gru_model.pt"
    },
    {
        "image_backbone": "efficientnet_b0",
        "audio_backbone": "lstm",
        "validation_mse": 2.787272185087204,
        "validation_mae": 1.5404645502567291,
        "test_mse": 2.901867628097534,
        "test_mae": 1.5843785762786866,
        "model_path": "test_models/efficientnet_b0_lstm_model.pt"
    },
    {
        "image_backbone": "resnet50",
        "audio_backbone": "gru",
        "validation_mse": 3.9335442543029786,
        "validation_mae": 1.8762320041656495,
        "test_mse": 3.72695152759552,
        "test_mae": 1.8381730556488036,
        "model_path": "test_models/resnet50_gru_model.pt"
    },
    {
        "image_backbone": "resnet50",
        "audio_backbone": "lstm",
        "validation_mse": 6.088638782501221,
        "validation_mae": 2.3887929677963258,
        "test_mse": 6.1847597599029545,
        "test_mae": 2.418113374710083,
        "model_path": "test_models/resnet50_lstm_model.pt"
    },
    {
        "image_backbone": "efficientnet_b3",
        "audio_backbone": "gru",
        "validation_mse": 104.58460273742676,
        "validation_mae": 10.183499813079834,
        "test_mse": 104.58482055664062,
        "test_mae": 10.180697345733643,
        "model_path": "test_models/efficientnet_b3_gru_model.pt"
    },
    {
        "image_backbone": "efficientnet_b3",
        "audio_backbone": "lstm",
        "validation_mse": 105.40057525634765,
        "validation_mae": 10.221695899963379,
        "test_mse": 105.17274551391601,
        "test_mae": 10.21053056716919,
        "model_path": "test_models/efficientnet_b3_lstm_model.pt"
    }
]