Skip to main content
SpringerLink
Log in

Fruit category classification via an eight-layer convolutional neural network with parametric rectified linear unit and dropout technique

  • Published:
Multimedia Tools and Applications Aims and scope Submit manuscript

Abstract

In this paper, we apply an improved deep convolutional neural network (CNN) in fruit category classification, which is a hotspot in computer vision field. We created an 8-layer deep convolutional neural network, and utilized parametric rectified linear unit to take the place of plain rectified linear unit, and placed dropout layer before each fully-connected layer. Data augmentation was used to help avoid overfitting. Our 8-layer deep convolutional neural network secured an overall accuracy of 95.67%. This proposed 8-layer method performs better than five state-of-the-art methods using traditional machine learning methods and one state-of-the-art CNN method.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Log in via an institution

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10
Fig. 11

Similar content being viewed by others

References

  1. Barushka A et al (2016) Spam filtering using regularized neural networks with rectified linear units. In: 15th International Conference of the Italian Association for Artificial Intelligence (AIIA). Springer Int Publishing Ag, Genova, p 65–75

  2. Benani A (2018) Fruits and vegetables: what is the real benefit? M S-Med Sci 34(4):316–317

    Google Scholar 

  3. Cheng H (2018) Classification of Alzheimer's disease based on eight-layer convolutional neural network with leaky rectified linear unit and max pooling. J Med Syst 42:42

    Google Scholar 

  4. Duggal R, et al (2017) P-TELU : parametric tan hyperbolic linear unit activation for deep neural networks. In: IEEE International Conference on Computer Vision Workshops. IEEE, Venice, p. 974–978

  5. He KM et al (2015) Delving deep into rectifiers: surpassing human-level performance on ImageNet classification. In: IEEE International Conference on Computer Vision (ICCV). IEEE, Santiago, p 1026–1034

  6. Ji G (2014) Fruit classification using computer vision and feedforward neural network. J Food Eng 143:167–177

    Google Scholar 

  7. Jiang Y (2018) Exploring a smart pathological brain detection method on pseudo Zernike moment. Multimedia Tools Appl 77(17):22589–22604

    Google Scholar 

  8. Jiang XH et al (2018) Deep neural networks with elastic rectified linear units for object recognition. Neurocomputing 275:1132–1139

    Google Scholar 

  9. Kamal-Eldin A et al (2018) Classification of date fruit (Phoenix dactylifera, L.) based on chemometric analysis with multivariate approach. Journal of Food Measurement and Characterization 12(2):1020–1027

    Google Scholar 

  10. Kheiralipour K et al (2017) Introducing new shape features for classification of cucumber fruit based on image processing technique and artificial neural networks. J Food Process Eng 40(6):4 Article ID. UNSP e12558

    Google Scholar 

  11. Li YA (2017) Fruit sensing and classification system by fractional fourier entropy and improved hybrid genetic algorithm. In: 5th International Conference on Industrial Application Engineering (IIAE). Kitakyushu, Japan, p 293–299

  12. Liu F et al (2017) Summary on fruit identification methods: a literature review. ESSAEME 119:1629–1633

    Google Scholar 

  13. Lu S (2018) Pathological brain detection in magnetic resonance imaging using combined features and improved extreme learning machines. J Med Imaging Health Inform 8:1486–1490

    Google Scholar 

  14. Lu SY et al (2016) Fruit classification by HPA-SLFN. In: 8th International Conference on Wireless Communications & Signal Processing. IEEE, Yangzhou, p. 11–17

  15. Muhammad K (2017) Image based fruit category classification by 13-layer deep convolutional neural network and data augmentation. Multimed Tools Appl. https://doi.org/10.1007/s11042-017-5243-3

  16. Pan C (2018) Multiple sclerosis identification by convolutional neural network with dropout and parametric ReLU. J Comput Sci 28:1–10

    MathSciNet  Google Scholar 

  17. Pan C (2018) Abnormal breast identification by nine-layer convolutional neural network with parametric rectified linear unit and rank-based stochastic pooling. J Comput Sci 27:57–68

    Google Scholar 

  18. Qian P (2018) Cat swarm optimization applied to alcohol use disorder identification. Multimedia Tools Appl 77(17):22875–22896

    Google Scholar 

  19. Qu YW (2013) Interacting multiple Gaussian particle filter. In: Hu T (ed) International conference on artificial intelligence and software engineering. Atlantis Press, Nanjing, pp 62–66

    Google Scholar 

  20. Qu YW et al (2008) Convergence properties of particle filter algorithm. In: Chinese Conference on Pattern Recognition. IEEE, Beijing, p 24–29

  21. Snetkov L et al (2018) An effective model based on Haar wavelet entropy and genetic algorithm for fruit identification. AIP Conf Proc 1955(1):040013

    Google Scholar 

  22. Srivastava N et al (2014) Dropout: a simple way to prevent neural networks from overfitting. J Mach Learn Res 15:1929–1958

    MathSciNet  MATH  Google Scholar 

  23. Sun J (2018) Preliminary study on angiosperm genus classification by weight decay and combination of most abundant color index with fractional Fourier entropy. Multimed Tools Appl 77(17):22671–22688

    Google Scholar 

  24. Tang C (2018) Twelve-layer deep convolutional neural network with stochastic pooling for tea category classification on GPU platform. Multimed Tools Appl 77(17):22821–22839

    Google Scholar 

  25. Wei G (2010) Color image enhancement based on HVS and PCNN. SCIENCE CHINA Inf Sci 53(10):1963–1976

    MathSciNet  Google Scholar 

  26. Wu LN (2008) Improved image filter based on SPCNN. Sci China Ser F-Inf Sci 51(12):2115–2125

    Google Scholar 

  27. Wu LN (2008) Pattern recognition via PCNN and Tsallis entropy. Sensors 8(11):7518–7529

    Google Scholar 

  28. Wu LN (2009) Segment-based coding of color images. Sci China Ser F-Inf Sci 52(6):914–925

    MATH  Google Scholar 

  29. Wu L (2011) Recursive structure element decomposition using migration fitness scaling genetic algorithm. In: 2nd International Conference on Swarm Intelligence (ICSI). Springer Verlag, Chongqing, p. 514–521

  30. Wu LN (2011) Bankruptcy prediction by genetic ant colony algorithm. In: International Conference on Computer-aided Material and Engineering. Trans Tech Publications Ltd., Hangzhou, p. 459–463

  31. Wu L (2012) Classification of fruits using computer vision and a multiclass support vector machine. Sensors 12(9):12489–12505

    Google Scholar 

  32. Wu J (2016) Fruit classification by biogeography-based optimization and feedforward neural network. Expert Syst 33(3):239–253

    Google Scholar 

  33. Zhang Z et al (2015) A survey of sparse representation: algorithms and applications. IEEE Access 3:490–530

    Google Scholar 

  34. Zhang Z et al (2017) Discriminative elastic-net regularized linear regression. IEEE Trans Image Process 26(3):1466–1481

    MathSciNet  MATH  Google Scholar 

  35. Zhao G (2018) Smart pathological brain detection by synthetic minority oversampling technique, extreme learning machine, and Jaya algorithm. Multimed Tools Appl 77(17):22629–22648

    Google Scholar 

Download references

Acknowledgements

This paper was funded by Open fund for Jiangsu Key Laboratory of Advanced Manufacturing Technology (HGAMTL-1703), Henan Key Research and Development Project (182102310629), National key research and development plan (2017YFB1103202), Natural Science Foundation of China (61602250).

Author information

Authors and Affiliations

  1. Department of Rheumatology and Immunology, Nanjing Drum Tower Hospital, The Affiliated Hospital of Nanjing University Medical School, Nanjing, China

    Shui-Hua Wang

  2. Department of Informatics, University of Leicester, Leicester, LE1 7RH, UK

    Shui-Hua Wang

  3. School of Architecture Building and Civil engineering, Loughborough University, Loughborough, LE11 3TU, UK

    Shui-Hua Wang

  4. School of Computer Science and Technology, Nanjing Normal University, Nanjing, 210008, Jiangsu, China

    Yi Chen

Authors
  1. Shui-Hua Wang
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. Yi Chen
    View author publications

    You can also search for this author in PubMed Google Scholar

Corresponding authors

Correspondence to Shui-Hua Wang or Yi Chen.

Ethics declarations

Conflict of interest

We have no conflicts of interest to disclose with regard to the subject matter of this paper.

Ethical approval

This article does not contain any studies with human participants or animals performed by any of the authors.

Additional information

Publisher’s Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Wang, SH., Chen, Y. Fruit category classification via an eight-layer convolutional neural network with parametric rectified linear unit and dropout technique. Multimed Tools Appl 79, 15117–15133 (2020). https://doi.org/10.1007/s11042-018-6661-6

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s11042-018-6661-6

Keywords

Search

Navigation