TRAINING AND TESTING DATASET python code for making the .pth CNN model (EfficientNet B0 by default) Steps Download the image dataset refer here Split the image dataset refer here Clone this repository and check the changeme at train_bruises.py and train_ripeness.py Run the command in CLI for the bruises python train_bruises.py for the ripeness python train_ripeness.py Other Remarks output graphs are in .png output CLIs are in .txt variable epoch=1 is default at main() img dataset path is set to be the same directory as the .py as seen on the .gitignore Timestamp for AlexNet