A simple playing card classifier implemented using the EfficientNet-B0 Model. A sample result after training the model is as shown:
After importing the dataset into the root directory, simply run main.py and wait for training to finish. Note that the training currently runs on a GPU on Mac, and the device must be changed if ran on another OS.
The dataset was obtained from Kaggle, in the link below: https://www.kaggle.com/datasets/gpiosenka/cards-image-datasetclassification