Add MobileViT-XXS 3D print failure detector + ONNX

README.md

@@ -44,5 +44,5 @@ print(label)
 ## Training Details
 - Base: `apple/mobilevit-xx-small` (1.3M params)
 - Image size: 256×256
-- Best val macro-F1: **0.9684**
+- Best val macro-F1: **0.9011**
 - Near-balanced dataset (normal≈3.7K / failure≈3.3K) — standard cross-entropy loss

model.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:610db62cb7dfbbfb4723e49ac683ff16aa23a1fcc695940571a4c1cb5d67f917
+oid sha256:6416a94903810842313614260ba6d6800edb284e4c080970cc395b917891debb
 size 4104856

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:605ee1c5062fcb7eccf1e20e6a8431f6ab730684cf6f7368ec9ec5f2efd065cf
+oid sha256:b85af008d63085d3e0eabd6e1b33a9e5f2ba47e960ebd6a4c3e7aaa7f736ecc9
 size 3869352

test_results.json

@@ -1,41 +1,41 @@
 {
   "classification_report": {
     "normal": {
-      "precision": 0.9696969696969697,
-      "recall": 0.9696969696969697,
-      "f1-score": 0.9696969696969697,
-      "support": 198.0
+      "precision": 0.9772727272727273,
+      "recall": 0.9297297297297298,
+      "f1-score": 0.9529085872576177,
+      "support": 185.0
     },
     "failure": {
-      "precision": 0.9302325581395349,
-      "recall": 0.9302325581395349,
-      "f1-score": 0.9302325581395349,
-      "support": 86.0
+      "precision": 0.8645833333333334,
+      "recall": 0.9540229885057471,
+      "f1-score": 0.907103825136612,
+      "support": 87.0
     },
-    "accuracy": 0.9577464788732394,
+    "accuracy": 0.9375,
     "macro avg": {
-      "precision": 0.9499647639182522,
-      "recall": 0.9499647639182522,
-      "f1-score": 0.9499647639182522,
-      "support": 284.0
+      "precision": 0.9209280303030303,
+      "recall": 0.9418763591177384,
+      "f1-score": 0.9300062061971148,
+      "support": 272.0
     },
     "weighted avg": {
-      "precision": 0.9577464788732394,
-      "recall": 0.9577464788732394,
-      "f1-score": 0.9577464788732394,
-      "support": 284.0
+      "precision": 0.9412286931818182,
+      "recall": 0.9375,
+      "f1-score": 0.9382577993733255,
+      "support": 272.0
     }
   },
   "confusion_matrix": {
     "layout": "[[TN, FP], [FN, TP]]",
     "values": [
       [
-        192,
-        6
+        172,
+        13
       ],
       [
-        6,
-        80
+        4,
+        83
       ]
     ]
   }

training_history.json

@@ -1,100 +1,37 @@
 [
   {
     "epoch": 1,
-    "tr_loss": 0.6544712557369884,
-    "tr_f1": 0.6847230631043242,
-    "vl_loss": 0.6045823617453289,
-    "vl_f1": 0.7589808027923212
+    "tr_loss": 0.6579452176788115,
+    "tr_f1": 0.6613059104360187,
+    "vl_loss": 0.6108984553506014,
+    "vl_f1": 0.7892942583732058
   },
   {
     "epoch": 2,
-    "tr_loss": 0.5423026512938551,
-    "tr_f1": 0.8040669914596561,
-    "vl_loss": 0.4713951907393789,
-    "vl_f1": 0.8212706833396488
+    "tr_loss": 0.5502447364710268,
+    "tr_f1": 0.8142255285369921,
+    "vl_loss": 0.481944231639489,
+    "vl_f1": 0.8623666835957339
   },
   {
     "epoch": 3,
-    "tr_loss": 0.389313331195106,
-    "tr_f1": 0.8634857989498674,
-    "vl_loss": 0.3226982529930007,
-    "vl_f1": 0.8792892967980115
+    "tr_loss": 0.4299551450481397,
+    "tr_f1": 0.8577008897442334,
+    "vl_loss": 0.3683563004560576,
+    "vl_f1": 0.8853799520654166
   },
   {
     "epoch": 4,
-    "tr_loss": 0.2852860887078498,
-    "tr_f1": 0.8973588514421574,
-    "vl_loss": 0.23448040412719173,
-    "vl_f1": 0.9390086206896552
+    "tr_loss": 0.35671244364683113,
+    "tr_f1": 0.8838141431895128,
+    "vl_loss": 0.32326523008381747,
+    "vl_f1": 0.901121563526183
   },
   {
     "epoch": 5,
-    "tr_loss": 0.21992481323423288,
-    "tr_f1": 0.9335041814003946,
-    "vl_loss": 0.19301387317504984,
-    "vl_f1": 0.9390086206896552
-  },
-  {
-    "epoch": 6,
-    "tr_loss": 0.1907757613290287,
-    "tr_f1": 0.9332493485973552,
-    "vl_loss": 0.1711783712933847,
-    "vl_f1": 0.9469043151969981
-  },
-  {
-    "epoch": 7,
-    "tr_loss": 0.1742152771696836,
-    "tr_f1": 0.9389346051086506,
-    "vl_loss": 0.15541251680349713,
-    "vl_f1": 0.9511341338984381
-  },
-  {
-    "epoch": 8,
-    "tr_loss": 0.15697269421029733,
-    "tr_f1": 0.9476696257016195,
-    "vl_loss": 0.14385232055566335,
-    "vl_f1": 0.9473194341027551
-  },
-  {
-    "epoch": 9,
-    "tr_loss": 0.14618295908999643,
-    "tr_f1": 0.9490398201669878,
-    "vl_loss": 0.13943202954193729,
-    "vl_f1": 0.9431489808848299
-  },
-  {
-    "epoch": 10,
-    "tr_loss": 0.15598203578366548,
-    "tr_f1": 0.9470454224842488,
-    "vl_loss": 0.1250224683710206,
-    "vl_f1": 0.9683874571132212
-  },
-  {
-    "epoch": 11,
-    "tr_loss": 0.1330390945898577,
-    "tr_f1": 0.9542768399185172,
-    "vl_loss": 0.12364656794092259,
-    "vl_f1": 0.9603351346300593
-  },
-  {
-    "epoch": 12,
-    "tr_loss": 0.12876383141899192,
-    "tr_f1": 0.9537759150347236,
-    "vl_loss": 0.1268359492713908,
-    "vl_f1": 0.9477216748768473
-  },
-  {
-    "epoch": 13,
-    "tr_loss": 0.12867094816933342,
-    "tr_f1": 0.9566096687330925,
-    "vl_loss": 0.11918994414458847,
-    "vl_f1": 0.9603351346300593
-  },
-  {
-    "epoch": 14,
-    "tr_loss": 0.127408712517337,
-    "tr_f1": 0.9521209973039881,
-    "vl_loss": 0.11911928340424076,
-    "vl_f1": 0.9560995284189625
+    "tr_loss": 0.3321886461404873,
+    "tr_f1": 0.8938713656122619,
+    "vl_loss": 0.3152350533712394,
+    "vl_f1": 0.901121563526183
   }
 ]

training_meta.json

@@ -5,9 +5,9 @@
   "batch_size": 64,
   "lr": 0.0003,
   "lr_backbone": 3e-05,
-  "weight_decay": 0.0001,
-  "num_epochs_trained": 14,
-  "best_val_macro_f1": 0.9684,
+  "weight_decay": 0.0002,
+  "num_epochs_trained": 5,
+  "best_val_macro_f1": 0.9011,
   "label2id": {
     "normal": 0,
     "failure": 1