It's a simple python project with a keras CNN model and a GUI interface for selecting a keras model and testing the model to classify an image
For training a new model you simply need to run the train_model.py file and if you want to change the settings for the training you need to go and modify it accordingly to your preferences. The model has now the a 80%+ accuracy and since I don't own a Nvidia GPU it takes a while to train the model for more than 50 epochs but I'm sure that running it for more epochs can improve the final accuracy.
I just started studying ML and Deep learning so the code itself is not perfect nor optimized to the fullest but feel free to improve it!