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