This repo is originally adapted from -
The Google Colab setup is available at -
https://colab.research.google.com/drive/1ldUUlBKpPg9aTmJg154J29KMLRt4HUxF?usp=sharing
In order to run the Colab notebook, the data and model files will also be required. All files are available at the below Drive location. This needs to be mounted in the Colab -
https://drive.google.com/drive/folders/1OjcrCeZmTdFccV7FCUJHbVVN2cdGPzqi?usp=sharing
Alternatively, you can download the folder from the Drive and run the script locally on a CUDA-enabled GPU.