Loading
Tem um workflow do ComfyUI? Gere uma documentação legível direto no navegador.Torne um workflow legível.Experimentar agora ↗
A from-scratch CNN that classifies 50x50 chess-board cells into 13 classes (empty + 6 white + 6 black pieces), trained on the synthetic koryakinp/chess-positions dataset with Kornia GPU augmentation and Optuna...
Fonte do modelo
Descrição da fonte
A from-scratch CNN that classifies 50x50 chess-board cells into 13 classes
(empty + 6 white + 6 black pieces), trained on the synthetic
koryakinp/chess-positions dataset with Kornia GPU augmentation and Optuna
hyperparameter tuning.
Fontes
1 fonteVerificado 28 de set.
Artefatos de modelo
1 artefatoTrechos de fonte
2 trechos--- license: cc0-1.0 tags: - image-classification - chess - pytorch --- # Chess Piece Classifier (Custom CNN) A from-scratch CNN that classifies 50x50 chess-board cells into 13 classes (empty + 6 white + 6 black pieces), trained on the synthetic `koryakinp/chess-positions` dataset with Kornia GPU augmentation and Optuna hyperparameter tuning. ## Classes - 0: empty - 1: wP - 2: wN - 3: wB - 4: wR - 5: wQ - 6: wK - 7: bP - 8: bN - 9: bB - 10: bR - 11: bQ - 12: bK ## Usage ```python import torch from huggingface_hub import hf_hub_download weights = hf_hub_download(repo_id="honi05/chess-piece-cnn", filename="chess_cnn.pt") state = torch.load(weights, map_location="cpu") ```
import torch
from huggingface_hub import hf_hub_download
weights = hf_hub_download(repo_id="honi05/chess-piece-cnn", filename="chess_cnn.pt")
state = torch.load(weights, map_location="cpu")
Source context: 0 downloads · 0 likes · Pipeline image-classification · Repo honi05/chess-piece-cnn