arunimas1107 commited on
Commit
24126e3
·
verified ·
1 Parent(s): f3e0a0d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +70 -0
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
+