Download the dataset here: https://drive.google.com/file/d/1oUzRXqhR1KGRNurcf4gew3JwHsZzb9CB/view?usp=sharing (I don't claim the ownership of this dataset)

Extract and copy folder dataset and place it in the root folder of this project

You can load the model with saved weights by download the weights here: https://drive.google.com/file/d/1vRFMmM2mUeYl-8MumYJdr38pFMDY_bv7/view?usp=sharing, and extract and copy folder checkpoints and place it in the root folder of this project

Run main.py file