Food101 classifier. Baseline, data augmentation, fine-tuning, hitting ~77% validation accuracy.
Computer vision classifier over the Food101 dataset, inspired by the original Food101 paper. It started with a simple CNN baseline that overfit quickly, added data augmentation (variance up, raw accuracy down, but the model generalizes better), moved to EfficientNet feature extraction, and finished by unfreezing the top layers for fine-tuning. Final validation accuracy was about 77%, with TensorBoard curves and confusion matrices tracking every step. Also tried EfficientNetB7 and documented that bigger wasn't better for this dataset. A useful "simple > complex" lesson captured in the writeup.
TensorFlowEfficientNetTransfer LearningData AugmentationTensorBoard
View on GitHub →