This code was implemented using the Colab.
To run it, simply upload the .ipynb file to Cobab and change the dataset path to your directory.
data_dir = '/path to your dataset.../'

test_data_dir = '/path to your test dataset.../'