Image based fruit category classification by 13-layer deep convolutional neural network and data augmentation

• Published in: Multimedia tools and applications Vol. 78; no. 3; pp. 3613 - 3632
• Main Authors: Zhang, Yu-Dong, Dong, Zhengchao, Chen, Xianqing, Jia, Wenjuan, Du, Sidan, Muhammad, Khan, Wang, Shui-Hua
• Format: Journal Article
• Language: English
• Published: New York Springer US 01.02.2019; Springer Nature B.V
• Subjects: Acceleration; Artificial neural networks; Computer Communication Networks; Computer Science; Computer vision; Data augmentation; Data Structures and Information Theory; Image classification; Image rotation; Industrial plants; Multimedia Information Systems; Neural networks; Special Purpose and Application-Based Systems; State of the art; Supermarkets; Vision systems; Fruit category identification; Softmax; Fully connected layer; Convolutional neural network
• ISSN: 1380-7501; 1573-7721
• DOI: 10.1007/s11042-017-5243-3

Fruit category identification is important in factories, supermarkets, and other fields. Current computer vision systems used handcrafted features, and did not get good results. In this study, our team designed a 13-layer convolutional neural network (CNN). Three types of data augmentation method was used: image rotation, Gamma correction, and noise injection. We also compared max pooling with average pooling. The stochastic gradient descent with momentum was used to train the CNN with minibatch size of 128. The overall accuracy of our method is 94.94%, at least 5 percentage points higher than state-of-the-art approaches. We validated this 13-layer is the optimal structure. The GPU can achieve a 177× acceleration on training data, and a 175× acceleration on test data. We observed using data augmentation can increase the overall accuracy. Our method is effective in image-based fruit classification.