Install all package requirements
- PyTorch (torch, torchvision, torchmetrics)
- AliveProgress
- Matplotlib
- Glob
Download the dataset from here: https://www.kaggle.com/datasets/gpiosenka/cards-image-datasetclassification
In Main.py: Set DATA_PATH to be the location for your data files.
Configure the following parameters at your risk:
- USE_AUGMENTED_PHOTOS
- PRINT_IMAGES (not supported)
- test_model (used to test saved model)
- continue_training (used to continue epochs on a saved model)
- configure the model layers that start around line 70
To Run: run main.py using python 3.11+