Improve card classification
Browse files- README.md +1 -1
- onnx/model.onnx +2 -2
- onnx/model.onnx.data +1 -1
- onnx/model_quantized.onnx +2 -2
README.md
CHANGED
|
@@ -15,7 +15,7 @@ ResNet18 ONNX model that classifies a cropped image of a SET® card into its fou
|
|
| 15 |
|
| 16 |
## Input / output
|
| 17 |
|
| 18 |
-
- Input: RGB card crop, resized to
|
| 19 |
- Output: four classification heads, in order count/color/fill/shape, each one of its three SET values plus `other` (low-confidence / non-card crop).
|
| 20 |
- `onnx/model.onnx` (fp32) and, if present, `onnx/model_quantized.onnx` (int8).
|
| 21 |
|
|
|
|
| 15 |
|
| 16 |
## Input / output
|
| 17 |
|
| 18 |
+
- Input: RGB card crop, resized to 112x160, normalized with ImageNet mean/std (`mean=[0.485, 0.456, 0.406]`, `std=[0.229, 0.224, 0.225]`).
|
| 19 |
- Output: four classification heads, in order count/color/fill/shape, each one of its three SET values plus `other` (low-confidence / non-card crop).
|
| 20 |
- `onnx/model.onnx` (fp32) and, if present, `onnx/model_quantized.onnx` (int8).
|
| 21 |
|
onnx/model.onnx
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a7ce4de08d94db7be5ff9cd1df8e0680a874ea6df0c2c38b45c95273c319a277
|
| 3 |
+
size 118518
|
onnx/model.onnx.data
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 44761088
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:94b7b00270c2c6cd2883cb1bee18afd9292c2aea6d3d8f51ef76a46d42d7cad9
|
| 3 |
size 44761088
|
onnx/model_quantized.onnx
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:984f690de4f073f81a7d9dec46e175a3f83d8e0b227c9672ab7d418dce331c51
|
| 3 |
+
size 11372823
|