Fruit Classification Based on Six Layer Convolutional Neural Network

Automatic fruit classification is a difficult problem because there are so many types of fruits and the large inter-class similarity. In this study, we proposed to use convolutional neural network (CNN) for fruit classification. We designed a six-layer CNN consisting of convolution layers, pooling layers and fully connected layers. The experiment results suggested that our method achieved promising performance with accuracy of 91.44%, better than three state-of-the-art approaches: voting-based support vector machine, wavelet entropy, and genetic algorithm.

[1]  Yudong Zhang,et al.  Fruit classification using computer vision and feedforward neural network , 2014 .

[2]  Toby P. Breckon,et al.  Using Deep Convolutional Neural Network Architectures for Object Classification and Detection Within X-Ray Baggage Security Imagery , 2018, IEEE Transactions on Information Forensics and Security.

[3]  Yang Li,et al.  Detection of Dendritic Spines Using Wavelet-Based Conditional Symmetric Analysis and Regularized Morphological Shared-Weight Neural Networks , 2015, Comput. Math. Methods Medicine.

[4]  Yudong Zhang,et al.  Identification of Green, Oolong and Black Teas in China via Wavelet Packet Entropy and Fuzzy Support Vector Machine , 2015, Entropy.

[5]  Siwalak Pathaveerat,et al.  Classification of longan fruit bruising using visible spectroscopy , 2011 .

[6]  Yudong Zhang,et al.  Dual-Tree Complex Wavelet Transform and Twin Support Vector Machine for Pathological Brain Detection , 2016 .

[7]  Yuankai Huo,et al.  FEATURE EXTRACTION OF BRAIN MRI BY STATIONARY WAVELET TRANSFORM AND ITS APPLICATIONS , 2010 .

[8]  Rafael Masot Peris,et al.  Odour sampling system with modifiable parameters applied to fruit classification , 2013 .

[9]  Yudong Zhang,et al.  Pathological brain detection based on wavelet entropy and Hu moment invariants. , 2015, Bio-medical materials and engineering.

[10]  Yudong Zhang,et al.  A Novel Compressed Sensing Method for Magnetic Resonance Imaging: Exponential Wavelet Iterative Shrinkage-Thresholding Algorithm with Random Shift , 2016, Int. J. Biomed. Imaging.

[11]  Yudong Zhang,et al.  Remote-Sensing Image Classification Based on an Improved Probabilistic Neural Network , 2009, Sensors.

[12]  Mahmoud Omid,et al.  An expert egg grading system based on machine vision and artificial intelligence techniques , 2013 .

[13]  Yudong Zhang,et al.  A novel algorithm for all pairs shortest path problem based on matrix multiplication and pulse coupled neural network , 2011, Digit. Signal Process..

[14]  Yudong Zhang,et al.  Classification of Fruits Using Computer Vision and a Multiclass Support Vector Machine , 2012, Sensors.

[15]  Yudong Zhang,et al.  Tea Category Classification Based on Feed-Forward Neural Network and Two-Dimensional Wavelet Entropy , 2015, HPCA.

[16]  Yudong Zhang,et al.  Crop Classification by Forward Neural Network with Adaptive Chaotic Particle Swarm Optimization , 2011, Sensors.

[17]  Yang Li,et al.  Hearing Loss Detection in Medical Multimedia Data by Discrete Wavelet Packet Entropy and Single-Hidden Layer Neural Network Trained by Adaptive Learning-Rate Back Propagation , 2017, ISNN.

[18]  Preetha Phillips,et al.  Leaf Recognition for Plant Classification Based on Wavelet Entropy and Back Propagation Neural Network , 2017, ICIRA.

[19]  Yudong Zhang,et al.  Feed‐forward neural network optimized by hybridization of PSO and ABC for abnormal brain detection , 2015, Int. J. Imaging Syst. Technol..

[20]  Wenjuan Jia,et al.  An effective model based on Haar wavelet entropy and genetic algorithm for fruit identification , 2018 .

[21]  Yudong Zhang,et al.  Fruit Classification by Wavelet-Entropy and Feedforward Neural Network Trained by Fitness-Scaled Chaotic ABC and Biogeography-Based Optimization , 2015, Entropy.

[22]  Preetha Phillips,et al.  Pathological brain detection in MRI scanning by wavelet packet Tsallis entropy and fuzzy support vector machine , 2015, SpringerPlus.

[23]  Yudong Zhang,et al.  Ford Motor Side-View Recognition System Based on Wavelet Entropy and Back Propagation Neural Network and Levenberg-Marquardt Algorithm , 2017, PAAP.

[24]  Sidan Du,et al.  Image based fruit category classification by 13-layer deep convolutional neural network and data augmentation , 2019, Multimedia Tools and Applications.

[25]  Yudong Zhang,et al.  A TS-PSO Based Artificial Neural Network for Short-Term Load Forecast , 2015, HPCA.

[26]  Yudong Zhang,et al.  Twelve-layer deep convolutional neural network with stochastic pooling for tea category classification on GPU platform , 2018, Multimedia Tools and Applications.

[27]  Yudong Zhang,et al.  Alcoholism Detection by Data Augmentation and Convolutional Neural Network with Stochastic Pooling , 2017, Journal of Medical Systems.

[28]  Yudong Zhang,et al.  Polarimetric synthetic aperture radar image segmentation by convolutional neural network using graphical processing units , 2017, Journal of Real-Time Image Processing.

[29]  Sidan Du,et al.  Cerebral Micro-Bleed Detection Based on the Convolution Neural Network With Rank Based Average Pooling , 2017, IEEE Access.

[30]  Ming Yang,et al.  Classification of Alzheimer’s Disease Based on Eight-Layer Convolutional Neural Network with Leaky Rectified Linear Unit and Max Pooling , 2018, Journal of Medical Systems.

[31]  Aijun Liu,et al.  Abnormal Breast Detection in Mammogram Images by Feed-forward Neural Network Trained by Jaya Algorithm , 2017, Fundam. Informaticae.