Update README.md
Browse files
README.md
CHANGED
|
@@ -1,7 +1,11 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
| 4 |
This model is to help determine the type of problem a 3D print has.
|
|
|
|
| 5 |
|
| 6 |
The model trained on images of 3D prints as they are printing as well as post printing.
|
| 7 |
Training set of images is about ~5GB
|
|
@@ -41,5 +45,4 @@ def preProcess(image):
|
|
| 41 |
|
| 42 |
input_image = transform(image).unsqueeze(0)
|
| 43 |
return input_image
|
| 44 |
-
```
|
| 45 |
-
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
+
language:
|
| 4 |
+
- en
|
| 5 |
+
pipeline_tag: image-to-text
|
| 6 |
---
|
| 7 |
This model is to help determine the type of problem a 3D print has.
|
| 8 |
+
The model uses AlexNet CNN Architecture built using PyTorch
|
| 9 |
|
| 10 |
The model trained on images of 3D prints as they are printing as well as post printing.
|
| 11 |
Training set of images is about ~5GB
|
|
|
|
| 45 |
|
| 46 |
input_image = transform(image).unsqueeze(0)
|
| 47 |
return input_image
|
| 48 |
+
```
|
|
|