ChessBoardDetector / README.md
honi05's picture
Add model card
53b9f74 verified
|
Raw
History Blame Contribute Delete
796 Bytes
---
license: mit
tags:
- computer-vision
- object-detection
- chess
- yolov11
- ultralytics
---
# 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
```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](https://github.com/Honi05/ChessBoardDetector)