wangtianthu/set-solver-models
By wangtianthu · ↓ 0 · ♥ 0 · Updated 2026-02-12
What's in this repo
This page lists the files in wangtianthu/set-solver-models, including best.onnx.
- 1 ONNX models (.onnx files)
- 15 files, including 7 .png, 2 .pt, 1 .md, 1 .json, 1 .yaml, 1 .csv
- Last updated 2026-02-12
ONNX files
| File | Size | Download |
|---|---|---|
| detector/weights/best.onnx | — | Download |
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
Set Solver Models
Trained models for the Set card game solver.
Live demo: huggingface.co/spaces/wangtianthu/set-solver
Models
Detector — YOLOv11n
Detects individual Set cards on a board image.
| Metric | Value |
|---|---|
| mAP50 | 99.5% |
| mAP50-95 | 97.4% |
| Architecture | YOLOv11n |
| Input size | 640x640 |
| Epochs | 10 |
| Training data | 4000 synthetic board images |
Files: detector/weights/best.pt (PyTorch), detector/weights/best.onnx (ONNX)
Classifier — MobileNetV3
Classifies each card's 4 attributes: shape, color, number, and fill.
| Metric | Value |
|---|---|
| Overall accuracy | 99.9% |
| Number accuracy | 100% |
| Color accuracy | 100% |
| Shape accuracy | 99.9% |
| Fill accuracy | 99.8% |
| Architecture | MobileNetV3-Small |
| Input size | 224x224 |
| Training data | ~9500 cropped card images (81 classes) |
File: classifier/classifier_best.pt
Usage
from ultralytics import YOLO
from PIL import Image
# Load detector
detector = YOLO("detector/weights/best.pt")
results = detector("board_photo.jpg", conf=0.25)
# Load classifier
import torch
from src.train.classifier import SetCardClassifier
classifier = SetCardClassifier(pretrained=False)
checkpoint = torch.load("classifier/classifier_best.pt", map_location="cpu")
classifier.load_state_dict(checkpoint["model_state_dict"])
classifier.eval()
Training
Both models were trained on synthetic data generated by a custom board generator that produces realistic Set game layouts with varied backgrounds, perspective transforms, and noise objects.
Source code: github.com/wangtian24/set-solver
Model info
Created: 2026-02-12
Updated: 2026-02-12
View on Hugging FaceFiles (15)
Related models
More from wangtianthu
No other models by this author.