Contains code for training a Unet model to segment chessboard images and a classification model to classify the pieces on the board.
This is a WIP, feedback welcome.
- Checkout this repo.
- Ensure submodules are checked out:
git submodule update --init - Create a virtual environment:
python -m venv .venv - Activate the virtual environment:
source .venv/bin/activate - Install dependencies:
pip install -r requirements.txt
See launch configurations in .vscode/launch.json for training the models.