For more information about the project details, see this blog post associated with this project.
- Python 3.5
- Keras 2.0
- pygame
- scikit-image
Clone the repository or download it. To test the pretrained model, run the "test.py" file. To retrain the model from scratch, run the "train_network.py" file and the trained models at different stages will be saved in "saved_models" folder.
This work is based on the following repos and blogs-