levfa
/

set-card-class

Image Classification

card-classification

Model card Files Files and versions

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.

Downloads last month: -; Downloads are not tracked for this model. How to track