Developed By: Mudit Chaudhary and Arjun Rao - The Chinese University of Hong Kong
- Food images classified into 75 "large categories" - each containing up to 1000 images each category.
- 25 small categories - containing 5 images per category
-
Data augmenation was used to enhance *Random rotations with range 40 degrees *Height and width shift *Zoom *Shear *Horizontal flip
-
Training Methodology:
- Train the ResNet-50 with images from the large sample
- Freeze the weights of that model and implement a new output layer for training the images from small sample.
- Train last few layers of the previous model.
- Transfer Learning from previous models
- Design methodology
For the first model: We add a Global Average Pooling Layer connecting ResNet. Add a Fully Connected layer with 1024 nodes Add a Dropout Layer with 40% dropout rate Add a Fully Connected output layer with 75 nodes For the second model: We add a Global Average Pooling Layer connecting ResNet. Add a Fully Connected layer with 1024 nodes Add a Dropout Layer with 40% dropout rate Add a Fully Connected output layer with 25 nodes
- Future plans : Future plan include: Tuning the hyperparameters Using Inception ResNet Modify the output layer to reduce overfitting for the small sample