-
Download the Jupyter notebook:
wget https://github.com/rezapace/Machine-Learning-chess-classification/releases/download/1.0/klasifikasi_catur.ipynb
-
Open Google Colab: https://colab.research.google.com/#create=true
-
Upload the downloaded
klasifikasi_catur.ipynbfile to Google Colab. -
Run the notebook:
- Execute each cell in order
- Follow the instructions provided in the notebook comments
Note: Make sure you have a Google account to use Google Colab. If you encounter any issues, please refer to the project repository for troubleshooting or to report problems.
This project implements a Convolutional Neural Network (CNN) to classify chess pieces. It uses TensorFlow and Keras to build and train a model that can identify five different chess pieces: king, queen, rook, bishop, and pawn.
- Image classification of chess pieces
- Utilizes Convolutional Neural Networks (CNN)
- Data augmentation for improved model performance
- Easy-to-use Google Colab implementation
- TensorFlow
- Keras
- scikit-learn
- Google Colab (for easy execution)
-
Download the Jupyter notebook file: klasifikasi_catur.ipynb
-
Open Google Colab
-
Upload the downloaded
.ipynbfile to Google Colab -
Run the following commands in a Colab cell to clone the repository and set up the environment:
!apt install git !git clone https://github.com/rezapace/Machine-Learning-chess-classification
-
Execute the cells in order:
- Import required libraries
- Prepare the dataset
- Set up data augmentation and loading
- Build the CNN model
- Compile the model
- Train the model
- Evaluate the model
- Save the model (optional)
-
The final cell will display the model's accuracy on the validation set.
The CNN model consists of:
- 3 Convolutional layers with ReLU activation
- 3 MaxPooling layers
- Flatten layer
- Dense layer with ReLU activation
- Dropout layer (0.5) for regularization
- Output layer with Softmax activation for 5 classes
The model achieves an accuracy of approximately 42.40% on the validation set after 10 epochs of training.
- Increase the dataset size
- Experiment with different model architectures
- Implement transfer learning using pre-trained models
- Fine-tune hyperparameters for better performance
Contributions to improve the model or extend the project are welcome. Please feel free to fork the repository and submit pull requests.
This project is open-source and available under the MIT License.
For more information or to report issues, please visit the project repository.