For more information about the project details, see this blog post associated with this project.
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-