Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
# Anomaly Detection Model – Edge AI for Casting Defect Inspection
|
| 2 |
|
| 3 |
## Overview
|
|
@@ -118,3 +119,72 @@ Applied AI Engineer
|
|
| 118 |
|
| 119 |
---
|
| 120 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
# Anomaly Detection Model – Edge AI for Casting Defect Inspection
|
| 3 |
|
| 4 |
## Overview
|
|
|
|
| 119 |
|
| 120 |
---
|
| 121 |
|
| 122 |
+
|
| 123 |
+
```yaml
|
| 124 |
+
|
| 125 |
+
model_name: "Anomaly Detection Model"
|
| 126 |
+
model_type: "autoencoder"
|
| 127 |
+
language: "en"
|
| 128 |
+
license: "mit"
|
| 129 |
+
|
| 130 |
+
tags:
|
| 131 |
+
- anomaly-detection
|
| 132 |
+
- autoencoder
|
| 133 |
+
- edge-ai
|
| 134 |
+
- openvino
|
| 135 |
+
- onnx
|
| 136 |
+
- computer-vision
|
| 137 |
+
- unsupervised-learning
|
| 138 |
+
|
| 139 |
+
task_categories:
|
| 140 |
+
- anomaly-detection
|
| 141 |
+
- image-classification
|
| 142 |
+
|
| 143 |
+
library_name: "pytorch"
|
| 144 |
+
|
| 145 |
+
datasets:
|
| 146 |
+
- name: "Casting Product Image Dataset"
|
| 147 |
+
source: "https://www.kaggle.com/datasets/ravirajsinh45/real-life-industrial-dataset-of-casting-product"
|
| 148 |
+
|
| 149 |
+
metrics:
|
| 150 |
+
- name: "Reconstruction Error Threshold"
|
| 151 |
+
type: "MSE"
|
| 152 |
+
value: 0.01
|
| 153 |
+
|
| 154 |
+
model-index:
|
| 155 |
+
- name: "Anomaly Detection Model"
|
| 156 |
+
results:
|
| 157 |
+
- task:
|
| 158 |
+
type: "anomaly-detection"
|
| 159 |
+
name: "Casting Defect Detection"
|
| 160 |
+
dataset:
|
| 161 |
+
name: "Casting Product Image Dataset"
|
| 162 |
+
type: "image"
|
| 163 |
+
metrics:
|
| 164 |
+
- name: "MSE Reconstruction Error"
|
| 165 |
+
type: "float"
|
| 166 |
+
value: 0.01
|
| 167 |
+
|
| 168 |
+
inference:
|
| 169 |
+
input_format: "Grayscale image (128x128)"
|
| 170 |
+
output_format: "Reconstructed image + anomaly score"
|
| 171 |
+
|
| 172 |
+
intended_use:
|
| 173 |
+
primary_use: "Industrial defect inspection via anomaly detection."
|
| 174 |
+
limitations:
|
| 175 |
+
- "Requires consistent lighting and background conditions."
|
| 176 |
+
- "Trained specifically on metal casting images."
|
| 177 |
+
|
| 178 |
+
author:
|
| 179 |
+
name: "Arunima Surendran"
|
| 180 |
+
role: "AI Developer & Researcher"
|
| 181 |
+
repository: "https://github.com/arunimakanavu/aonmalydetectionmodel"
|
| 182 |
+
email: "N/A"
|
| 183 |
+
|
| 184 |
+
framework_versions:
|
| 185 |
+
pytorch: "2.2.0"
|
| 186 |
+
openvino: "2024.1"
|
| 187 |
+
onnx: "1.15.0"
|
| 188 |
+
|
| 189 |
+
```
|
| 190 |
+
|