Junk Food Recognition Neural Network

To train/test this model download and extract here dataset folder. Link in overview file.

Files:

• model.py - model class
• datamodule.py - dataset / transforms
• test_model.py - testing model on validation set
• train.py - model training
• overview.ipynb - description of dataset, model, dataloader, optimizer etc. + model testing + simple classifier