proj-airi/games-balatro-2024-yolo-card-identity-detection-mod-ground-truth

<p align="center"><img src="./docs/cover.png"></p>

Balatro card identity detection

YOLO11n detector for 54 exact card identities, card backs, and unreadable rank/suit. This model is part of the Project AIRI game-playing work.

Files

• —model.pt: Ultralytics/PyTorch checkpoint.
• —onnx/model.onnx: fixed-shape ONNX export at 960 px.
• —docs/: training curves, confusion matrices, and prediction examples.

Independent test split

The train/validation/test split is grouped by generated game round. Metrics describe the Mod-generated distribution and are not a claim of identical accuracy on arbitrary live captures.

Prediction examples

Training diagnostics

[image]

[image]

License

MIT.