Skip to content

Onigiri-Capstone/ml-model-training

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 

Repository files navigation


Onigiri Restaurant Recommendation

Food Recognition Model

Built With

Data Exploration & Preparation

We use Food Images dataset, food images collection from various countries including Indonesia, China, India, South Korea, Thailand, Japan. This dataset has ~200 images in each class.

This how our dataset looks like:

Dataset

The dataset we use is obtained from scrapping process using bing-image-downloader. The dataset we got from bing scrapping will be divided into train, validation, and test data.

You can see the detail of our scrapping dataset process here.

Technique Used

Images Classification

We use TensorFlow API to do Transfer Learning on images clasification task for our custom food images dataset. We use MobileNetV2 model as base model, do fine tuning, and add some output layers at the bottom of the pretrained model. MobileNetV2 model, which previously trained on `Imagenet` dataset, has some base of knowledge that will help us in classifying foods task. At the end, our model will produce food class predictions with the highest probability score.

After we trained the model on ~5000 images of Indonesian Restaurants' common foods, we got 81% validation accuracy.

This is our training result:

Training Result

And this is our learning curve:

Learning Curve

You can see the detail of our training process and dataset that we use here.

Acknowledgements

About

ML model for food classification task.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published