metadata
license: mit
tags:
- chess
- object-detection
- yolo
- fen
- ocr
library_name: ultralytics
SmallVision — 2D Chess OCR Models
Finetuned YOLO and RF-DETR weights for detecting chessboards, pieces, orientation markers, and last-move tiles in 2D chess screenshots.
Derived from AndrewSpano/2d-chess-ocr.
Files
| File | Backend | Size |
|---|---|---|
yolo26m-finetuned.pt |
YOLO | ~42 MB |
yolo26m-finetuned.onnx |
YOLO ONNX | ~78 MB |
yolo26n-finetuned.pt |
YOLO | ~5 MB |
yolo26n-finetuned.onnx |
YOLO ONNX | ~10 MB |
rfdetr-l-finetuned.pth |
RF-DETR-L | ~129 MB |
rfdetr-2xl-finetuned.pth |
RF-DETR-2XL | ~484 MB |
Usage
from huggingface_hub import hf_hub_download
path = hf_hub_download(
repo_id="scchess/smallvision",
filename="yolo26m-finetuned.pt",
repo_type="model",
)
See SmallVision for the full inference workflow.