| license: agpl-3.0 | |
| library_name: onnx | |
| tags: | |
| - image-classification | |
| - resnet18 | |
| - multi-task | |
| - card-classification | |
| pipeline_tag: image-classification | |
| # set-card-class | |
| ResNet18 ONNX model that classifies a cropped image of a SET® card into its four attributes: count, color, fill, shape. | |
| ## Input / output | |
| - Input: RGB card crop, resized to 224x320, normalized with ImageNet mean/std (`mean=[0.485, 0.456, 0.406]`, `std=[0.229, 0.224, 0.225]`). | |
| - Output: four classification heads, in order count/color/fill/shape, each one of its three SET values plus `other` (low-confidence / non-card crop). | |
| - `onnx/model.onnx` (fp32) and, if present, `onnx/model_quantized.onnx` (int8). | |
| ## License | |
| AGPL-3.0. | |
| ## Disclaimer | |
| Unofficial, fan-made project; not affiliated with, endorsed by, or sponsored by SET Enterprises, Inc. | |