Introduction

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.

Getting Started

• 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

Training

See launch configurations in .vscode/launch.json for training the models.