-
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