File size: 1,622 Bytes
202ce04 4c1de52 202ce04 4c1de52 40afab2 4c1de52 40afab2 4c1de52 40afab2 4c1de52 40afab2 4c1de52 202ce04 4c1de52 | 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 | {
"labels": [
"barn",
"bridge",
"castle",
"mosque",
"skyscraper",
"stadium",
"temple",
"windmill"
],
"num_classes": 8,
"label_to_idx": {
"barn": 0,
"bridge": 1,
"castle": 2,
"mosque": 3,
"skyscraper": 4,
"stadium": 5,
"temple": 6,
"windmill": 7
},
"idx_to_label": {
"0": "barn",
"1": "bridge",
"2": "castle",
"3": "mosque",
"4": "skyscraper",
"5": "stadium",
"6": "temple",
"7": "windmill"
},
"model_file": "fine_tuning_swa.keras",
"input_shape": [
320,
320,
3
],
"preprocessing": "efficientnet_v2.preprocess_input",
"architecture": "EfficientNetV2-S(ImageNet,frozen_Ph1)+Conv2D(256,3x3,relu,same)+BN+MaxPooling2D(2,2)+GeMPooling(p=3.0,trainable)+Dense(256,relu)+BN+Dropout(0.4)+Dense(n_classes,softmax)",
"training": {
"phase1": {
"optimizer": "AdamW",
"lr": 0.001,
"weight_decay": 2e-05,
"warmup_epochs": 3,
"max_epochs": 25,
"es_patience": 7,
"augmentation": "Mixup+CutMix+standard",
"focal_loss_gamma": 2.0,
"label_smoothing": 0.1
},
"phase2": {
"optimizer": "DiscriminativeAdamW",
"lr": 0.0003,
"weight_decay": 2e-05,
"lr_multipliers": {
"block6": 0.1
},
"warmup_epochs": 5,
"max_epochs": 50,
"es_patience": 12,
"augmentation": "standard_only",
"focal_loss_gamma": 2.0,
"label_smoothing": 0.05,
"unfreeze": "block6+top_conv (BN frozen)"
},
"swa": {
"epochs": 10,
"lr": 0.0001,
"bn_update_steps": 100
}
}
} |