The checkpoints folder is the v1 version of the model.
v1 model specs:
- 54 countries, south korea a lot
- Generally trained on 350 images per country, no transformations.
- Super overfitted to training dataset.
v2 model specs:
- Removed uganda
- Added horizontal flipping, and some random zoom to the training phase.
- Trained on approx 1000 images per country
To run on screen:
poetry run python run.py --data data --model mode-v2