This is a mini-experiment conducted after finishing Lectures 01 and 02 of the FastAI course. The goal is to classify images of Sushi, Steak, and Pizza.
The dataset used for fine-tuning is pizza_steak_sushi.zip from the PyTorch Tutorial by Daniel Bourke.
Fine-tuning the pretrained Resnet34 model for only 3 epochs surprisingly gave quite accurate results in a very short time.
Check the last section of FoodClassifier.ipynb.