A Vision-Based Method Utilizing Deep Convolutional Neural Networks for Fruit Variety Classification in Uncertainty Conditions of Retail Sales

This study proposes a double-track method for the classification of fruit varieties for application in retail sales. The method uses two nine-layer Convolutional Neural Networks (CNNs) with the same architecture, but different weight matrices. The first network classifies fruits according to images of fruits with a background, and the second network classifies based on images with the ROI (Region Of Interest, a single fruit). The results are aggregated with the proposed values of weights (importance). Consequently, the method returns the predicted class membership with the Certainty Factor (CF). The use of the certainty factor associated with prediction results from the original images and cropped ROIs is the main contribution of this paper. It has been shown that CFs indicate the correctness of the classification result and represent a more reliable measure compared to the probabilities on the CNN outputs. The method is tested with a dataset containing images of six apple varieties. The overall image classification accuracy for this testing dataset is excellent (99.78%). In conclusion, the proposed method is highly successful at recognizing unambiguous, ambiguous, and uncertain classifications, and it can be used in a vision-based sales systems in uncertain conditions and unplanned situations.

[1]  Rafflesia Khan,et al.  Multi Class Fruit Classification Using Efficient Object Detection and Recognition Techniques , 2019, International Journal of Image, Graphics and Signal Processing.

[2]  A. Geetha,et al.  Fruits, Vegetable and Plants Category Recognition Systems Using Convolutional Neural Networks : A Review , 2019, International Journal of Scientific Research in Computer Science, Engineering and Information Technology.

[3]  Naeem Hussain,et al.  Intra-Class Recognition of Fruits Using DCNN for Commercial Trace Back-System , 2019, Proceedings of the 2019 4th International Conference on Multimedia Systems and Signal Processing - ICMSSP 2019.

[4]  En Li,et al.  Detection of Apple Lesions in Orchards Based on Deep Learning Methods of CycleGAN and YOLOV3-Dense , 2019, J. Sensors.

[5]  Zaidah Ibrahim,et al.  Comparing bags of features, conventional convolutional neural network and AlexNet for fruit recognition , 2019, Indonesian Journal of Electrical Engineering and Computer Science.

[6]  En Li,et al.  Apple detection during different growth stages in orchards using the improved YOLO-V3 model , 2019, Comput. Electron. Agric..

[7]  Ghulam Muhammad,et al.  Automatic Fruit Classification Using Deep Learning for Industrial Applications , 2019, IEEE Transactions on Industrial Informatics.

[8]  Douglas Chai,et al.  A comprehensive review of fruit and vegetable classification techniques , 2018, Image Vis. Comput..

[9]  Fernando Alonso-Fernandez,et al.  Fruit and Vegetable Identification Using Machine Learning for Retail Applications , 2018, 2018 14th International Conference on Signal-Image Technology & Internet-Based Systems (SITIS).

[10]  Ling Wang,et al.  Design of fruits and vegetables online inspection system based on vision , 2018 .

[11]  Qianhua He,et al.  Automatic Fruit Recognition Based on DCNN for Commercial Source Trace System , 2018, International Journal on Computational Science & Applications.

[12]  Mark Sandler,et al.  MobileNetV2: Inverted Residuals and Linear Bottlenecks , 2018, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[13]  Mihai Oltean,et al.  Fruit recognition from images using deep learning , 2017, Acta Universitatis Sapientiae, Informatica.

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

[15]  Masayu Leylia Khodra,et al.  Toward a new approach in fruit recognition using hybrid RGBD features and fruit hierarchy property , 2017, 2017 4th International Conference on Electrical Engineering, Computer Science and Informatics (EECSI).

[16]  Meng-Meng Yang,et al.  A study on classification of fruit type and fruit disease , 2017 .

[17]  Р Ю Чуйков,et al.  Обнаружение транспортных средств на изображениях загородных шоссе на основе метода Single shot multibox Detector , 2017 .

[18]  Frans Coenen,et al.  Traffic sign recognition with convolutional neural network based on max pooling positions , 2016, 2016 12th International Conference on Natural Computation, Fuzzy Systems and Knowledge Discovery (ICNC-FSKD).

[19]  Leonard Barolli,et al.  A Vegetable Category Recognition System Using Deep Neural Network , 2016, 2016 10th International Conference on Innovative Mobile and Internet Services in Ubiquitous Computing (IMIS).

[20]  F. Garcia,et al.  Fruit Classification by Extracting Color Chromaticity, Shape and Texture Features: Towards an Application for Supermarkets , 2016, IEEE Latin America Transactions.

[21]  Yudong Zhang,et al.  Fruit classification by biogeography‐based optimization and feedforward neural network , 2016, Expert Syst. J. Knowl. Eng..

[22]  Yi Li,et al.  R-FCN: Object Detection via Region-based Fully Convolutional Networks , 2016, NIPS.

[23]  Wei Liu,et al.  SSD: Single Shot MultiBox Detector , 2015, ECCV.

[24]  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.

[25]  Kaiming He,et al.  Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks , 2015, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[26]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

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

[28]  Nitish Srivastava,et al.  Dropout: a simple way to prevent neural networks from overfitting , 2014, J. Mach. Learn. Res..

[29]  Yann LeCun,et al.  What is the best multi-stage architecture for object recognition? , 2009, 2009 IEEE 12th International Conference on Computer Vision.

[30]  Gabriel Taubin,et al.  VeggieVision: a produce recognition system , 1996, Proceedings Third IEEE Workshop on Applications of Computer Vision. WACV'96.