NutriNet: A Deep Learning Food and Drink Image Recognition System for Dietary Assessment

被引:161
作者
Mezgec, Simon [1 ]
Seljak, Barbara Korousic [2 ]
机构
[1] Jozef Stefan Int Postgrad Sch, Informat & Commun Technol, Jamova Cesta 39, Ljubljana 1000, Slovenia
[2] Jozef Stefan Inst, Comp Syst Dept, Jamova Cesta 39, Ljubljana 1000, Slovenia
基金
欧盟地平线“2020”;
关键词
NutriNet; deep convolutional neural networks; deep learning; food recognition; food detection; drink recognition; drink detection; Parkinson's disease;
D O I
10.3390/nu9070657
中图分类号
R15 [营养卫生、食品卫生]; TS201 [基础科学];
学科分类号
100403 ;
摘要
Automatic food image recognition systems are alleviating the process of food-intake estimation and dietary assessment. However, due to the nature of food images, their recognition is a particularly challenging task, which is why traditional approaches in the field have achieved a low classification accuracy. Deep neural networks have outperformed such solutions, and we present a novel approach to the problem of food and drink image detection and recognition that uses a newly-defined deep convolutional neural network architecture, called NutriNet. This architecture was tuned on a recognition dataset containing 225,953 512 x 512 pixel images of 520 different food and drink items from a broad spectrum of food groups, on which we achieved a classification accuracy of 86.72%, along with an accuracy of 94.47% on a detection dataset containing 130,517 images. We also performed a real-world test on a dataset of self-acquired images, combined with images from Parkinson's disease patients, all taken using a smartphone camera, achieving a top-five accuracy of 55%, which is an encouraging result for real-world images. Additionally, we tested NutriNet on the University of Milano-Bicocca 2016 (UNIMIB2016) food image dataset, on which we improved upon the provided baseline recognition result. An online training component was implemented to continually fine-tune the food and drink recognition model on new images. The model is being used in practice as part of a mobile app for the dietary assessment of Parkinson's disease patients.
引用
收藏
页数:19
相关论文
共 34 条
[1]  
[Anonymous], P IEEE INT C COMP VI
[2]  
[Anonymous], PROC CVPR IEEE
[3]  
[Anonymous], OVERVIEW GRADIENT DE
[4]  
[Anonymous], 2016, ENET DEEP NEURAL NET
[5]  
[Anonymous], 2015, PROC CVPR IEEE
[6]  
[Anonymous], 2014, P 2014 ACM INT JOINT
[7]  
[Anonymous], P MM 14 ORL FL US 3
[8]  
[Anonymous], 2013, INT C LEARN REPR ICL
[9]  
[Anonymous], 1983, SOV MATH DOKL
[10]  
[Anonymous], 2013, Some improvements on deep convolutional neural network based image classification