Machine Learning Based Approach on Food Recognition and Nutrition Estimation
Zhidong Shen, Adnan Shehzad, Si Chen, Hui Sun, Jin Liu
Abstract
Nowadays, standard intake of healthy food is necessary for keeping a balanced diet to avoid obesity in the human body. In this paper, we present a novel system based on machine learning that automatically performs accurate classification of food images and estimates food attributes. This paper proposes a deep learning model consisting of a convolutional neural network that classifies food into specific categories in the training part of the prototype system. The main purpose of the proposed method is to improve the accuracy of the pre-training model. The paper designs a prototype system based on the client server model. The client sends an image detection request and processes it on the server side. The prototype system is designed with three main software components, including a pre-trained CNN model training module for classification purposes, a text data training module for attribute estimation models, and a server-side module. We experimented with a variety of food categories, each containing thousands of images, and through machine learning training to achieve higher classification accuracy.