CoolFace
Modelpublic

honi05/ChessBoardDetector

sourceHugging Facemitupdated 3mo agoView on Hugging Face
0likes334downloads
Model Card

ChessBoardDetector

Two-model YOLOv11 pipeline that converts a chessboard photograph into a FEN string.

Models

FileArchitectureTask
board_detector.ptYOLOv11nLocate the chessboard bounding box
piece_detector.ptYOLOv11sIdentify all 12 piece types on a 512×512 rectified board

Usage

python
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

GitHub — honi05/ChessBoardDetector