Update README.md
Browse files
README.md
CHANGED
|
@@ -138,72 +138,3 @@ Applied AI Engineer
|
|
| 138 |
[GitHub Repository](https://github.com/arunimakanavu/anomalydetectionmodel)
|
| 139 |
|
| 140 |
---
|
| 141 |
-
|
| 142 |
-
|
| 143 |
-
```yaml
|
| 144 |
-
|
| 145 |
-
model_name: "Anomaly Detection Model"
|
| 146 |
-
model_type: "autoencoder"
|
| 147 |
-
language: "en"
|
| 148 |
-
license: "mit"
|
| 149 |
-
|
| 150 |
-
tags:
|
| 151 |
-
- anomaly-detection
|
| 152 |
-
- autoencoder
|
| 153 |
-
- edge-ai
|
| 154 |
-
- openvino
|
| 155 |
-
- onnx
|
| 156 |
-
- computer-vision
|
| 157 |
-
- unsupervised-learning
|
| 158 |
-
|
| 159 |
-
task_categories:
|
| 160 |
-
- anomaly-detection
|
| 161 |
-
- image-classification
|
| 162 |
-
|
| 163 |
-
library_name: "pytorch"
|
| 164 |
-
|
| 165 |
-
datasets:
|
| 166 |
-
- name: "Casting Product Image Dataset"
|
| 167 |
-
source: "https://www.kaggle.com/datasets/ravirajsinh45/real-life-industrial-dataset-of-casting-product"
|
| 168 |
-
|
| 169 |
-
metrics:
|
| 170 |
-
- name: "Reconstruction Error Threshold"
|
| 171 |
-
type: "MSE"
|
| 172 |
-
value: 0.01
|
| 173 |
-
|
| 174 |
-
model-index:
|
| 175 |
-
- name: "Anomaly Detection Model"
|
| 176 |
-
results:
|
| 177 |
-
- task:
|
| 178 |
-
type: "anomaly-detection"
|
| 179 |
-
name: "Casting Defect Detection"
|
| 180 |
-
dataset:
|
| 181 |
-
name: "Casting Product Image Dataset"
|
| 182 |
-
type: "image"
|
| 183 |
-
metrics:
|
| 184 |
-
- name: "MSE Reconstruction Error"
|
| 185 |
-
type: "float"
|
| 186 |
-
value: 0.01
|
| 187 |
-
|
| 188 |
-
inference:
|
| 189 |
-
input_format: "Grayscale image (128x128)"
|
| 190 |
-
output_format: "Reconstructed image + anomaly score"
|
| 191 |
-
|
| 192 |
-
intended_use:
|
| 193 |
-
primary_use: "Industrial defect inspection via anomaly detection."
|
| 194 |
-
limitations:
|
| 195 |
-
- "Requires consistent lighting and background conditions."
|
| 196 |
-
- "Trained specifically on metal casting images."
|
| 197 |
-
|
| 198 |
-
author:
|
| 199 |
-
name: "Arunima Surendran"
|
| 200 |
-
role: "AI Developer & Researcher"
|
| 201 |
-
repository: "https://github.com/arunimakanavu/aonmalydetectionmodel"
|
| 202 |
-
email: "N/A"
|
| 203 |
-
|
| 204 |
-
framework_versions:
|
| 205 |
-
pytorch: "2.2.0"
|
| 206 |
-
openvino: "2024.1"
|
| 207 |
-
onnx: "1.15.0"
|
| 208 |
-
|
| 209 |
-
```
|
|
|
|
| 138 |
[GitHub Repository](https://github.com/arunimakanavu/anomalydetectionmodel)
|
| 139 |
|
| 140 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|