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 Jun 26

Commit

a09a1b5

·

verified ·

1 Parent(s): 40afab2

Upload model_benchmark.json with huggingface_hub

Files changed (1) hide show

model_benchmark.json +21 -21

model_benchmark.json CHANGED Viewed

@@ -1,32 +1,32 @@
 {
-  "model_file": "best_phase2_swa.keras",
-  "architecture": "DenseNet121+Conv2D(256)+BN+MaxPool2D+GeMPooling+Dense(256)+BN+Dropout(0.4)+Dense(8,softmax)",
-  "total_params": 9466953,
-  "trainable_params": 7884297,
-  "non_trainable_params": 1582656,
   "input_shape": [
     320,
     320,
     3
   ],
   "num_classes": 8,
-  "model_size_mb": 97.78,
-  "saved_model_size_mb": 76.29,
-  "tflite_size_kb": 36726.35,
-  "tfjs_size_mb": 36.5,
-  "inference_ms_keras": 365.3,
-  "inference_ms_tflite": 157.9,
   "metrics": {
     "train_accuracy": 0.9988,
-    "val_accuracy": 0.9658,
-    "test_accuracy": 0.9688,
-    "test_loss": 0.4485,
-    "tta_accuracy": 0.968,
-    "top1_accuracy": 0.9688,
-    "top2_accuracy": 0.9859,
-    "top3_accuracy": 0.9933,
-    "macro_auc": 0.9986,
-    "ece": 0.1438,
-    "overfitting_gap": 0.03
   }
 }

 {
+  "model_file": "fine_tuning_swa.keras",
+  "architecture": "EfficientNetV2-S+Conv2D(256)+BN+MaxPool2D+GeMPooling+Dense(256)+BN+Dropout(0.4)+Dense(8,softmax)",
+  "total_params": 23350633,
+  "trainable_params": 17810225,
+  "non_trainable_params": 5540408,
   "input_shape": [
     320,
     320,
     3
   ],
   "num_classes": 8,
+  "model_size_mb": 226.75,
+  "saved_model_size_mb": 183.29,
+  "tflite_size_kb": 90483.55,
+  "tfjs_size_mb": 89.54,
+  "inference_ms_keras": 358.0,
+  "inference_ms_tflite": 170.0,
   "metrics": {
     "train_accuracy": 0.9988,
+    "val_accuracy": 0.9836,
+    "test_accuracy": 0.9777,
+    "test_loss": 0.4262,
+    "tta_accuracy": 0.9799,
+    "top1_accuracy": 0.9777,
+    "top2_accuracy": 0.9926,
+    "top3_accuracy": 0.997,
+    "macro_auc": 0.9985,
+    "ece": 0.1204,
+    "overfitting_gap": 0.0211
   }
 }