Upload AutoLens EfficientNet-B2 artifact

README.md

@@ -0,0 +1,77 @@
+---
+library_name: onnxruntime
+pipeline_tag: image-classification
+tags:
+  - computer-vision
+  - image-classification
+  - cars
+  - onnx
+  - safetensors
+license: other
+---
+
+# AutoLens EfficientNet-B2 Candidate
+
+Current strongest CNN candidate for the AutoLens AI demo artifact lane. This repository is part of the AutoLens AI Yazlab 2 project.
+
+## Model
+
+- Architecture: EfficientNet-B2
+- Classes: 8 vehicle body types
+- Source run: `baseline_0_efficientnet_b2_20260509_135313`
+- Source checkpoint: `checkpoints/baseline_0_efficientnet_b2_20260509_135313/best-04-0.8994.ckpt`
+- Export format: `model.safetensors` plus `metadata.json`, with ONNX for deployment
+
+## Classes
+
+- SUV
+- VAN
+- STATION WAGON
+- MICRO
+- OPEN WHEEL / F1
+- SEDAN
+- HATCHBACK
+- PICK UP
+
+## Preprocessing
+
+- RGB input
+- Resize: 256
+- Center crop / model input: 224 x 224
+- Mean: `[0.4429, 0.4354, 0.437]`
+- Std: `[0.2456, 0.2421, 0.2449]`
+
+## Calibration
+
+Post-hoc temperature scaling was fitted on the validation split only.
+
+- Temperature: `1.779860`
+- Validation samples: 3157
+- NLL: 0.29796 -> 0.24160
+- ECE: 0.03823 -> 0.01043
+- Accuracy: 0.92398
+
+No internal test or instructor final test images were used for calibration.
+
+## Artifact sizes
+
+- `model.safetensors`: 29.722 MB
+- `model.onnx`: 29.371 MB
+- Project limit: 95 MB
+
+## Files
+
+- `model.safetensors` — weights-only model artifact
+- `metadata.json` — architecture, classes, preprocessing, source run, and artifact metadata
+- `model.onnx` — ONNX Runtime inference artifact
+- `model.simplified.onnx` — optional simplified ONNX graph when available
+- `calibration.json` — temperature scaling metadata
+- `size_report.json` / `size_check.json` — artifact size and ONNX smoke evidence
+
+## Intended use
+
+Educational demo and report evidence for classifying uploaded vehicle images into the 8 AutoLens body-type classes.
+
+## Limitations
+
+The dataset is assembled from public/open sources and may contain domain bias. Similar body styles such as hatchback, station wagon, and sedan can be ambiguous. The merged raw dataset is not redistributed in this model repository.

calibration.json

@@ -0,0 +1,23 @@
+{
+  "created_at": "2026-05-09T15:20:01.700775+00:00",
+  "method": "temperature_scaling",
+  "temperature": 1.779860258102417,
+  "fit_split": "artifacts/dataset/splits/val.csv",
+  "forbidden_splits": [
+    "internal_test",
+    "instructor_final_test"
+  ],
+  "num_samples": 3157,
+  "before": {
+    "nll": 0.2979576885700226,
+    "ece": 0.03823123127222061,
+    "accuracy": 0.9239784479141235,
+    "mean_confidence": 0.9612730741500854
+  },
+  "after": {
+    "nll": 0.24159657955169678,
+    "ece": 0.01043415255844593,
+    "accuracy": 0.9239784479141235,
+    "mean_confidence": 0.9190037250518799
+  }
+}

metadata.json

@@ -0,0 +1,108 @@
+{
+  "artifact_format_version": 1,
+  "created_at": "2026-05-09T15:19:12.150358+00:00",
+  "source": {
+    "checkpoint_path": "checkpoints/baseline_0_efficientnet_b2_20260509_135313/best-04-0.8994.ckpt",
+    "checkpoint_sha256": "d54d26116c4a246f3a01eef941069d627432547c7be90bd5efa31d81ff43eb8c",
+    "run_id": "baseline_0_efficientnet_b2_20260509_135313",
+    "config_path": "configs/experiments/baseline_0_efficientnet_b2.yaml",
+    "metrics_csv": "checkpoints/baseline_0_efficientnet_b2_20260509_135313/metrics.csv",
+    "git": {
+      "commit": "d38a2a6f16b5579a0e0c6dd07ea4bb2c5c087429",
+      "dirty": "true"
+    }
+  },
+  "model": {
+    "architecture": "timm",
+    "model_name": "efficientnet_b2",
+    "num_classes": 8,
+    "pretrained_at_training": true,
+    "use_lora": false,
+    "lora_r": 8,
+    "lora_alpha": 16,
+    "lora_dropout": 0.1,
+    "lora_target_modules": null
+  },
+  "classes": {
+    "labels": [
+      "SUV",
+      "VAN",
+      "STATION WAGON",
+      "MICRO",
+      "OPEN WHEEL / F1",
+      "SEDAN",
+      "HATCHBACK",
+      "PICK UP"
+    ],
+    "class_to_idx": {
+      "SUV": 0,
+      "VAN": 1,
+      "STATION WAGON": 2,
+      "MICRO": 3,
+      "OPEN WHEEL / F1": 4,
+      "SEDAN": 5,
+      "HATCHBACK": 6,
+      "PICK UP": 7
+    },
+    "display_labels": [
+      "SUV",
+      "VAN",
+      "STATION WAGON",
+      "MICRO",
+      "OPEN WHEEL / F1",
+      "SEDAN",
+      "HATCHBACK",
+      "PICK UP"
+    ]
+  },
+  "preprocessing": {
+    "image_size": 224,
+    "resize_size": 256,
+    "crop_size": 224,
+    "mean": [
+      0.4429,
+      0.4354,
+      0.437
+    ],
+    "std": [
+      0.2456,
+      0.2421,
+      0.2449
+    ],
+    "color_mode": "RGB"
+  },
+  "data": {
+    "train_csv": "artifacts/dataset/splits/train.csv",
+    "val_csv": "artifacts/dataset/splits/val.csv",
+    "test_csv_reserved_not_used_for_calibration": "artifacts/dataset/splits/internal_test.csv",
+    "data_root": "datasets"
+  },
+  "training": {
+    "epoch": 4,
+    "global_step": 1980,
+    "learning_rate": 0.001,
+    "weight_decay": 0.0001,
+    "label_smoothing": 0.0,
+    "metrics_summary": {
+      "epoch": 17,
+      "test/acc": 0.9205047488212585,
+      "test/f1_macro": 0.9003689289093018,
+      "test/f1_weighted": 0.920499324798584,
+      "test/loss": 0.28352469205856323
+    }
+  },
+  "artifacts": {
+    "directory": "artifacts/export/efficientnet_b2_latest",
+    "safetensors": "artifacts/export/efficientnet_b2_latest/model.safetensors",
+    "metadata": "artifacts/export/efficientnet_b2_latest/metadata.json",
+    "onnx": "artifacts/export/efficientnet_b2_latest/model.onnx",
+    "calibration": "artifacts/export/efficientnet_b2_latest/calibration.json",
+    "safetensors_sha256": "1ac7acaa040e0a944d9deb2314067d7034b6e7460f3e30b79e8c03280ab423d4",
+    "safetensors_size_bytes": 31166288,
+    "size_report": "artifacts/export/efficientnet_b2_latest/size_report.json"
+  },
+  "license_notes": {
+    "base_model": "efficientnet_b2 pretrained backbone; verify upstream license before public publishing.",
+    "dataset": "Do not publish merged raw dataset images; document source links/counts instead."
+  }
+}

model.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9980bc05cfa3b3e6688001bb659ff9146c19b978ea6e106b008f745990cbefa1
+size 30797958

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1ac7acaa040e0a944d9deb2314067d7034b6e7460f3e30b79e8c03280ab423d4
+size 31166288

model.simplified.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3476de4b63f25ecc1648a216c3bd5d3c8352f0369c221cdf462355a8a3c30000
+size 30826165

size_check.json

@@ -0,0 +1,18 @@
+{
+  "max_size_mb": 95.0,
+  "results": [
+    {
+      "path": "artifacts/export/efficientnet_b2_latest/model.safetensors",
+      "exists": true,
+      "size_mb": 29.722,
+      "under_limit": true
+    },
+    {
+      "path": "artifacts/export/efficientnet_b2_latest/model.onnx",
+      "exists": true,
+      "size_mb": 29.371,
+      "under_limit": true
+    }
+  ],
+  "all_under_limit": true
+}

size_report.json

@@ -0,0 +1,33 @@
+{
+  "max_size_mb": 95.0,
+  "artifacts": {
+    "safetensors": {
+      "path": "artifacts/export/efficientnet_b2_latest/model.safetensors",
+      "size_mb": 29.722,
+      "under_limit": true
+    },
+    "onnx": {
+      "path": "artifacts/export/efficientnet_b2_latest/model.onnx",
+      "size_mb": 29.371,
+      "under_limit": true
+    },
+    "simplified_onnx": {
+      "path": "artifacts/export/efficientnet_b2_latest/model.simplified.onnx",
+      "size_mb": 29.398,
+      "under_limit": true
+    }
+  },
+  "onnxruntime_smoke": {
+    "provider": "CPUExecutionProvider",
+    "input_name": "image",
+    "output_shape": [
+      1,
+      8
+    ]
+  },
+  "simplification": {
+    "attempted": true,
+    "ok": true,
+    "path": "artifacts/export/efficientnet_b2_latest/model.simplified.onnx"
+  }
+}