Image Classification
Keras
LiteRT
TF-Keras
Safetensors
English
efficientnetv2-s
efficientnetv2
fgic
transfer-learning
gem-pooling
focal-loss
swa
grad-cam
calibration
temperature-scaling
computer-vision
tensorflow.js
Eval Results (legacy)
Instructions to use 0xgr3y/Arch-Building-Image-Classification with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Keras
How to use 0xgr3y/Arch-Building-Image-Classification with Keras:
# Available backend options are: "jax", "torch", "tensorflow". import os os.environ["KERAS_BACKEND"] = "jax" import keras model = keras.saving.load_model("hf://0xgr3y/Arch-Building-Image-Classification") - Notebooks
- Google Colab
- Kaggle
Upload config.json with huggingface_hub
Browse files- config.json +42 -42
config.json
CHANGED
|
@@ -58,30 +58,30 @@
|
|
| 58 |
"optimizer": "AdamW",
|
| 59 |
"lr": 0.001,
|
| 60 |
"epochs_max": 25,
|
| 61 |
-
"epochs_actual":
|
| 62 |
-
"val_accuracy": 0.
|
| 63 |
-
"val_loss":
|
| 64 |
-
"train_accuracy": 0.
|
| 65 |
"cutmix_mixup": true,
|
| 66 |
"label_smoothing": 0.1,
|
| 67 |
-
"early_stop_reason": "
|
| 68 |
},
|
| 69 |
{
|
| 70 |
"name": "Fase 2 - Selective Fine-Tuning",
|
| 71 |
"optimizer": "DiscriminativeAdamW",
|
| 72 |
"lr": 0.0003,
|
| 73 |
"epochs_max": 50,
|
| 74 |
-
"epochs_actual":
|
| 75 |
-
"val_accuracy": 0.
|
| 76 |
-
"val_loss": 0.
|
| 77 |
-
"train_accuracy": 0.
|
| 78 |
"unfreeze": "block6+top_conv (BN frozen)",
|
| 79 |
"discriminative_lr": {
|
| 80 |
"block6": 0.1
|
| 81 |
},
|
| 82 |
"cutmix_mixup": false,
|
| 83 |
"label_smoothing": 0.05,
|
| 84 |
-
"early_stop_reason": "
|
| 85 |
},
|
| 86 |
{
|
| 87 |
"name": "SWA Post-Training",
|
|
@@ -89,62 +89,62 @@
|
|
| 89 |
"lr": 0.0001,
|
| 90 |
"bn_update": true,
|
| 91 |
"bn_update_steps": 100,
|
| 92 |
-
"val_accuracy": 0.
|
| 93 |
-
"val_loss": 0.
|
| 94 |
"method": "Izmailov et al., UAI 2018"
|
| 95 |
}
|
| 96 |
],
|
| 97 |
"metrics": {
|
| 98 |
-
"train_accuracy": 0.
|
| 99 |
-
"val_accuracy": 0.
|
| 100 |
-
"test_accuracy": 0.
|
| 101 |
-
"test_loss": 0.
|
| 102 |
-
"tta_accuracy": 0.
|
| 103 |
-
"overfitting_gap": 0.
|
| 104 |
-
"test_correct":
|
| 105 |
"test_total": 1344,
|
| 106 |
-
"macro_precision": 0.
|
| 107 |
-
"macro_recall": 0.
|
| 108 |
-
"macro_f1": 0.
|
| 109 |
"per_class_f1": {
|
| 110 |
"barn": 0.9731,
|
| 111 |
-
"bridge": 0.
|
| 112 |
-
"castle": 0.
|
| 113 |
-
"mosque": 0.
|
| 114 |
-
"skyscraper": 0.
|
| 115 |
-
"stadium": 0.
|
| 116 |
-
"temple": 0.
|
| 117 |
-
"windmill": 0.
|
| 118 |
},
|
| 119 |
"per_class_recall": {
|
| 120 |
"barn": 0.9702,
|
| 121 |
-
"bridge": 0.
|
| 122 |
-
"castle": 0.
|
| 123 |
"mosque": 0.9821,
|
| 124 |
"skyscraper": 0.994,
|
| 125 |
"stadium": 0.9762,
|
| 126 |
-
"temple": 0.
|
| 127 |
"windmill": 0.9881
|
| 128 |
},
|
| 129 |
"checkpoint_comparison": {
|
| 130 |
"fine_tuning_swa": {
|
| 131 |
-
"val_accuracy": 0.
|
| 132 |
-
"val_loss": 0.
|
| 133 |
"rank": 0
|
| 134 |
},
|
| 135 |
-
"
|
| 136 |
-
"val_accuracy": 0.
|
| 137 |
-
"val_loss": 0.
|
| 138 |
"rank": 1
|
| 139 |
},
|
| 140 |
-
"
|
| 141 |
-
"val_accuracy": 0.
|
| 142 |
-
"val_loss": 0.
|
| 143 |
"rank": 2
|
| 144 |
},
|
| 145 |
"head_training": {
|
| 146 |
-
"val_accuracy": 0.
|
| 147 |
-
"val_loss":
|
| 148 |
"rank": 3
|
| 149 |
}
|
| 150 |
}
|
|
|
|
| 58 |
"optimizer": "AdamW",
|
| 59 |
"lr": 0.001,
|
| 60 |
"epochs_max": 25,
|
| 61 |
+
"epochs_actual": 25,
|
| 62 |
+
"val_accuracy": 0.9747,
|
| 63 |
+
"val_loss": 0.6859,
|
| 64 |
+
"train_accuracy": 0.861,
|
| 65 |
"cutmix_mixup": true,
|
| 66 |
"label_smoothing": 0.1,
|
| 67 |
+
"early_stop_reason": "EarlyStopping val_accuracy plateau"
|
| 68 |
},
|
| 69 |
{
|
| 70 |
"name": "Fase 2 - Selective Fine-Tuning",
|
| 71 |
"optimizer": "DiscriminativeAdamW",
|
| 72 |
"lr": 0.0003,
|
| 73 |
"epochs_max": 50,
|
| 74 |
+
"epochs_actual": 7,
|
| 75 |
+
"val_accuracy": 0.9777,
|
| 76 |
+
"val_loss": 0.3988,
|
| 77 |
+
"train_accuracy": 0.9722,
|
| 78 |
"unfreeze": "block6+top_conv (BN frozen)",
|
| 79 |
"discriminative_lr": {
|
| 80 |
"block6": 0.1
|
| 81 |
},
|
| 82 |
"cutmix_mixup": false,
|
| 83 |
"label_smoothing": 0.05,
|
| 84 |
+
"early_stop_reason": "EarlyStopping val_accuracy plateau"
|
| 85 |
},
|
| 86 |
{
|
| 87 |
"name": "SWA Post-Training",
|
|
|
|
| 89 |
"lr": 0.0001,
|
| 90 |
"bn_update": true,
|
| 91 |
"bn_update_steps": 100,
|
| 92 |
+
"val_accuracy": 0.9851,
|
| 93 |
+
"val_loss": 0.3638,
|
| 94 |
"method": "Izmailov et al., UAI 2018"
|
| 95 |
}
|
| 96 |
],
|
| 97 |
"metrics": {
|
| 98 |
+
"train_accuracy": 0.9997,
|
| 99 |
+
"val_accuracy": 0.9851,
|
| 100 |
+
"test_accuracy": 0.9792,
|
| 101 |
+
"test_loss": 0.3928,
|
| 102 |
+
"tta_accuracy": 0.9814,
|
| 103 |
+
"overfitting_gap": 0.0206,
|
| 104 |
+
"test_correct": 1316,
|
| 105 |
"test_total": 1344,
|
| 106 |
+
"macro_precision": 0.9794,
|
| 107 |
+
"macro_recall": 0.9792,
|
| 108 |
+
"macro_f1": 0.9792,
|
| 109 |
"per_class_f1": {
|
| 110 |
"barn": 0.9731,
|
| 111 |
+
"bridge": 0.9763,
|
| 112 |
+
"castle": 0.9762,
|
| 113 |
+
"mosque": 0.9851,
|
| 114 |
+
"skyscraper": 0.9853,
|
| 115 |
+
"stadium": 0.982,
|
| 116 |
+
"temple": 0.9818,
|
| 117 |
+
"windmill": 0.9736
|
| 118 |
},
|
| 119 |
"per_class_recall": {
|
| 120 |
"barn": 0.9702,
|
| 121 |
+
"bridge": 0.9821,
|
| 122 |
+
"castle": 0.9762,
|
| 123 |
"mosque": 0.9821,
|
| 124 |
"skyscraper": 0.994,
|
| 125 |
"stadium": 0.9762,
|
| 126 |
+
"temple": 0.9643,
|
| 127 |
"windmill": 0.9881
|
| 128 |
},
|
| 129 |
"checkpoint_comparison": {
|
| 130 |
"fine_tuning_swa": {
|
| 131 |
+
"val_accuracy": 0.9851,
|
| 132 |
+
"val_loss": 0.3638,
|
| 133 |
"rank": 0
|
| 134 |
},
|
| 135 |
+
"fine_tuning_ema": {
|
| 136 |
+
"val_accuracy": 0.9784,
|
| 137 |
+
"val_loss": 0.3732,
|
| 138 |
"rank": 1
|
| 139 |
},
|
| 140 |
+
"fine_tuning": {
|
| 141 |
+
"val_accuracy": 0.9777,
|
| 142 |
+
"val_loss": 0.3988,
|
| 143 |
"rank": 2
|
| 144 |
},
|
| 145 |
"head_training": {
|
| 146 |
+
"val_accuracy": 0.9747,
|
| 147 |
+
"val_loss": 0.6859,
|
| 148 |
"rank": 3
|
| 149 |
}
|
| 150 |
}
|