metadata

library_name: ultralytics
tags:
  - object-detection
  - chess
  - computer-vision
  - yolo
datasets:
  - chess-pieces
pipeline_tag: object-detection

Chess Piece Detection Model

This is a YOLO model trained to detect chess pieces on a chessboard.

Model Details

Model Type: YOLOv8/YOLOv11 Object Detection
Task: Chess piece detection and classification
Framework: Ultralytics YOLO
Repository: dopaul/chessboard-segmentation

Files

The following files are included in this model:

best.pt

Usage

from ultralytics import YOLO

# Load the model
model = YOLO('path/to/best.pt')

# Run inference
results = model('path/to/chess_image.jpg')

# Display results
results[0].show()

Model Performance

This model can detect and classify various chess pieces including:

Pawns
Rooks
Knights
Bishops
Queens
Kings

For both black and white pieces.

Training Data

The model was trained on chess piece datasets to achieve robust detection across different chess sets and lighting conditions.