Automated fruit recognition using EfficientNet and MixNet

被引:92
作者
Duong, Linh T. [1 ]
Nguyen, Phuong T. [2 ]
Di Sipio, Claudio [2 ]
Di Ruscio, Davide [2 ]
机构
[1] Duy Tan Univ, Inst Res & Dev, Da Nang, Vietnam
[2] Univ Aquila, Dept Informat Engn Comp Sci & Math, I-67100 Laquila, Italy
基金
欧盟地平线“2020”;
关键词
MACHINE LEARNING ALGORITHMS; AGRICULTURE;
D O I
10.1016/j.compag.2020.105326
中图分类号
S [农业科学];
学科分类号
09 ;
摘要
The classification of fruits offers many useful applications in daily life, such as automated harvesting or building up stocks for supermarkets. Studies have been proposed to classify fruits from input images, exploiting image processing and machine learning techniques. Though a lot of improvements have been achieved in recent years, many approaches still suffer prolonged training/testing time, or a considerably high number of false positives. For several applications, it is crucial to provide users with not only precise but also real-time recommendations. In this paper, we propose a practical solution to fruit recognition by exploiting two recently-developed classifiers that have demonstrated themselves to be both effective and efficient. We adopted EfficientNet and MixNet, two families of deep neural networks to build an expert system being able to accurately and swiftly identify fruits. Such a system can be deployed onto devices with limited computational resources to prompt exact and timely recommendations. The approach's performance has been validated on a real dataset consisting of 48,905 images for training and 16,421 images for testing. The experimental results showed that the application of EfficientNet and MixNet on the considered dataset substantially improves the overall prediction accuracy in comparison to a well-established baseline.
引用
收藏
页数:10
相关论文
共 31 条
[1]   Semi-supervised learning with convolutional neural networks for UAV images automatic recognition [J].
Amorim, Willian Paraguassu ;
Tetila, Everton Castelao ;
Pistori, Hemerson ;
Papa, Joao Paulo .
COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2019, 164
[2]  
[Anonymous], 2016, CLEF
[3]  
[Anonymous], 2017, 5 INT C LEARN REPR I
[4]  
Bishop C.M., 1995, Neural networks for pattern recognition
[5]  
Cai H., 2019, PROC INT C LEARN REP
[6]   A Few Useful Things to Know About Machine Learning [J].
Domingos, Pedro .
COMMUNICATIONS OF THE ACM, 2012, 55 (10) :78-87
[7]   Deep learning models for plant disease detection and diagnosis [J].
Ferentinos, Konstantinos P. .
COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2018, 145 :311-318
[8]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[9]  
Howard A. G., 2017, CoRR
[10]   Transfer Learning with Deep Convolutional Neural Network for SAR Target Classification with Limited Labeled Data [J].
Huang, Zhongling ;
Pan, Zongxu ;
Lei, Bin .
REMOTE SENSING, 2017, 9 (09)