0xgr3y
/

Arch-Building-Image-Classification

Image Classification

efficientnetv2-s

transfer-learning

temperature-scaling

computer-vision

Eval Results (legacy)

Model card Files Files and versions

0xgr3y commited on 12 days ago

Commit

40afab2

·

verified ·

1 Parent(s): 735a201

Upload label_mapping.json with huggingface_hub

Files changed (1) hide show

label_mapping.json +5 -5

label_mapping.json CHANGED Viewed

@@ -30,14 +30,14 @@
     "6": "temple",
     "7": "windmill"
   },
-  "model_file": "best_phase2_swa.keras",
   "input_shape": [
     320,
     320,
     3
   ],
-  "preprocessing": "densenet121.preprocess_input",
-  "architecture": "DenseNet121(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",
@@ -55,7 +55,7 @@
       "lr": 0.0003,
       "weight_decay": 2e-05,
       "lr_multipliers": {
-        "conv4_block": 0.1
       },
       "warmup_epochs": 5,
       "max_epochs": 50,
@@ -63,7 +63,7 @@
       "augmentation": "standard_only",
       "focal_loss_gamma": 2.0,
       "label_smoothing": 0.05,
-      "unfreeze": "conv4_block+conv5_block (BN frozen)"
     },
     "swa": {
       "epochs": 10,

     "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.0003,
       "weight_decay": 2e-05,
       "lr_multipliers": {
+        "block6": 0.1
       },
       "warmup_epochs": 5,
       "max_epochs": 50,
       "augmentation": "standard_only",
       "focal_loss_gamma": 2.0,
       "label_smoothing": 0.05,
+      "unfreeze": "block6+top_conv (BN frozen)"
     },
     "swa": {
       "epochs": 10,