SmallVision — 2D Chess OCR Models
Canonical smallchess org repo for finetuned YOLO and RF-DETR weights used by SmallVision.
Detects chessboards, pieces, orientation markers, and last-move tiles in 2D chess screenshots.
Derived from AndrewSpano/2d-chess-ocr.
Previously mirrored from the user repo
scchess/smallvision. Usesmallchess/smallvisiongoing forward.
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="smallchess/smallvision",
filename="yolo26m-finetuned.pt",
repo_type="model",
)
CLI:
hf download smallchess/smallvision --repo-type model --local-dir ./finetuned-models
See SmallVision for the full inference workflow.
- Downloads last month
- 62
# Couldn't find a valid YOLO version tag. # Replace XX with the correct version. from ultralytics import YOLOvXX model = YOLOvXX.from_pretrained("smallchess/smallvision") source = 'http://images.cocodataset.org/val2017/000000039769.jpg' model.predict(source=source, save=True)