Create folder models
Create folder datasets
Create folder original inside datasets
Download https://drive.google.com/open?id=1PZWa6H0i1PCH9zuYcIh5Ouk_p-9Gh58B, extract and place on datasets/original folder
Then, you can run python train.py to train and python test.py to test