Downloads · 30 days
369
24% of all-time downloads
dopaul/chessboard-detector
chessboard-detector is a object detection model from dopaul. Use it when you need objects located in an image. It is set up for ultralytics.
This is a YOLO model trained to detect chess pieces on a chessboard.
Downloads · 30 days
369
24% of all-time downloads
All-time downloads
1.5K
Public
Repo size
10.9 MB
Likes
1
Public
Click a slice to open those files.
.pt5.5 MB · 100%
From the Hugging Face model README
This is a YOLO model trained to detect chess pieces on a chessboard.
The following files are included in this model:
best.ptfrom 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()
This model can detect and classify various chess pieces including:
For both black and white pieces.
The model was trained on chess piece datasets to achieve robust detection across different chess sets and lighting conditions.