ChessBoardDetector
Two-model YOLOv11 pipeline that converts a chessboard photograph into a FEN string.
Models
| File | Architecture | Task |
|---|---|---|
board_detector.pt |
YOLOv11n | Locate the chessboard bounding box |
piece_detector.pt |
YOLOv11s | Identify all 12 piece types on a 512×512 rectified board |
Usage
from src.pipeline import run_pipeline
import cv2
image = cv2.imread("photo.jpg")
fen, annotated = run_pipeline(
image,
board_model_path="board_detector.pt",
piece_model_path="piece_detector.pt",
)
print(fen)
Source
- Downloads last month
- 189
from ultralytics import YOLOvv11 model = YOLOvv11.from_pretrained("honi05/ChessBoardDetector") source = 'http://images.cocodataset.org/val2017/000000039769.jpg' model.predict(source=source, save=True)