Initial FactoryVision AI model package (private review)

README.md

@@ -0,0 +1,53 @@
+---
+language:
+- en
+pipeline_tag: image-classification
+library_name: pytorch
+tags:
+- industrial-vision
+- factoryvision-ai
+- casting-defect-detection
+- education
+---
+# FactoryVision AI — Custom CNN for Casting Inspection
+
+**FactoryVision AI — Credits and Attribution**
+
+**Original Developer & Model Author:** Thevanth Rajamohan  
+**Technical Review, Code Contributions & Model Publication Support:** Swapin Vidya  
+**Publishing Organisation:** TechSphere Consultancy Services Private Limited  
+**Hugging Face:** https://huggingface.co/TechsphereAI  
+**Official Website:** https://techsphereconsultancy.com/
+
+**Project Provenance:** FactoryVision AI was originally developed by Thevanth Rajamohan. Swapin Vidya contributed technical review, detection-metrics improvements, testing, model publication preparation and documentation. The models are being prepared for publication under TechSphere Consultancy Services Private Limited's Hugging Face namespace, with the original developer's permission.
+
+**Original GitHub Repository:** https://github.com/Theivanth/Factory-Vision-AI  
+**Merged contribution by Swapin Vidya:** https://github.com/Theivanth/Factory-Vision-AI/pull/1
+
+## Purpose and limitations
+Educational two-class classification of casting images (`def_front` = defective; `ok_front` = acceptable). For **human-reviewed prototype inspection only**, not an industrial release decision or certified quality system.
+
+## Architecture and files
+- Custom PyTorch CNN trained from scratch; see the source repository's `src/quality.py` for architecture and preprocessing
+- Weights: `model.pth` (archive `checkpoints/casting_cnn.pth`)
+- Exact ordered labels in `labels.json`
+- Use the project's original loader and preprocessing rather than assuming a generic Hugging Face Transformer loader can use the `.pth` file.
+
+## Training dataset
+Real-Life Industrial Dataset of Casting Product by Ravirajsinh Dabhi (https://www.kaggle.com/datasets/ravirajsinh45/real-life-industrial-dataset-of-casting-product). The source README identifies its Kaggle licence as **CC BY-NC-ND 4.0**. It includes non-commercial and no-derivatives restrictions. The rights to publish dataset-trained checkpoint weights require separate verification; model-developer consent does not settle third-party licensing. No input images are redistributed here.
+
+## Held-out test performance (from supplied Kaggle output)
+- Accuracy: **0.9860**
+- Macro precision: **0.9849**
+- Macro recall: **0.9849**
+- Macro F1: **0.9849**
+- Test confusion matrix (rows true, columns predicted; class order `def_front`, `ok_front`): `[[448, 5], [5, 257]]`
+- Mean inference latency reported by Kaggle: **0.608 ms** (environment/measurement specific; not a target CPU benchmark)
+
+The metrics are reported, not independently reproduced. The model is not validated on new factory deployments.
+
+## Inference and dependencies
+The checkpoint is intended to be loaded using the matching architecture from `src/quality.py` in https://github.com/Theivanth/Factory-Vision-AI . Do **not** call arbitrary `torch.load` on untrusted checkpoints. The project has known image-size, normalisation, and class-order requirements; check the current source for exact implementation before inference.
+
+## Licensing and distribution
+The source project has an MIT licence for **its code**, not necessarily the checkpoint. **Keep this repository private until the casting dataset's non-commercial/no-derivatives conditions and trained-weight publication rights have been resolved.** No model-weight licence is asserted in this card.

SOURCE.md

@@ -0,0 +1,8 @@
+Source: https://github.com/Theivanth/Factory-Vision-AI
+Original Project Developer and Model Author: Thevanth Rajamohan
+Technical Reviewer, Code Contributor and Hugging Face Publication Support: Swapin Vidya
+Publishing Organisation: TechSphere Consultancy Services Private Limited
+Hugging Face: https://huggingface.co/TechsphereAI
+Official Website: https://techsphereconsultancy.com/
+Swapin Vidya's merged contribution: https://github.com/Theivanth/Factory-Vision-AI/pull/1
+Model checkpoint SHA-256: 572501b104e64582206938e8f4038735ab082b8d4e2fe473b47ffa15aeffc427

labels.json

@@ -0,0 +1,4 @@
+[
+  "def_front",
+  "ok_front"
+]

metrics.json

@@ -0,0 +1,69 @@
+{
+  "checkpoint": "casting_cnn.pth",
+  "validation": {
+    "accuracy": 0.9954337899543378,
+    "precision_macro": 0.9953362800030287,
+    "recall_macro": 0.9953362800030287,
+    "macro_f1": 0.9953362800030287,
+    "per_class_precision": {
+      "def_front": 0.9960106382978723,
+      "ok_front": 0.994661921708185
+    },
+    "per_class_recall": {
+      "def_front": 0.9960106382978723,
+      "ok_front": 0.994661921708185
+    },
+    "per_class_f1": {
+      "def_front": 0.9960106382978723,
+      "ok_front": 0.994661921708185
+    },
+    "confusion_matrix": [
+      [
+        749,
+        3
+      ],
+      [
+        3,
+        559
+      ]
+    ],
+    "classes": [
+      "def_front",
+      "ok_front"
+    ],
+    "mean_inference_latency_ms": 0.6061720205487975
+  },
+  "test": {
+    "accuracy": 0.986013986013986,
+    "precision_macro": 0.9849392514702662,
+    "recall_macro": 0.9849392514702662,
+    "macro_f1": 0.9849392514702662,
+    "per_class_precision": {
+      "def_front": 0.9889624724061811,
+      "ok_front": 0.9809160305343512
+    },
+    "per_class_recall": {
+      "def_front": 0.9889624724061811,
+      "ok_front": 0.9809160305343512
+    },
+    "per_class_f1": {
+      "def_front": 0.9889624724061811,
+      "ok_front": 0.9809160305343512
+    },
+    "confusion_matrix": [
+      [
+        448,
+        5
+      ],
+      [
+        5,
+        257
+      ]
+    ],
+    "classes": [
+      "def_front",
+      "ok_front"
+    ],
+    "mean_inference_latency_ms": 0.6078423272728369
+  }
+}

model.pth

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