---
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.