Enhancing deep convolutional neural network models for orange quality classification using MobileNetV2 and data augmentation techniques

被引:0
|
作者
Huong, Phan Thi [1 ]
Hien, Lam Thanh [1 ]
Son, Nguyen Minh [1 ]
Tuan, Huynh Cao [1 ]
Nguyen, Thanh Q. [2 ]
机构
[1] Lac Hong Univ, Fac Informat Technol, Bien Hoa, Vietnam
[2] Nguyen Tat Thanh Univ, Inst Interdisciplinary Social Sci, 300A Nguyen Tat Thanh St,Ward 13,Dist 4, Ho Chi Minh City 700000, Vietnam
关键词
Deep convolutional neural network; MobileNetV2; data augmentation; feature extraction; depthwise separable convolutions; linear bottlenecks; inverted residuals; image classification;
D O I
10.1177/17483026241309070
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
This study introduces significant improvements in the construction of deep convolutional neural network models for classifying agricultural products, specifically oranges, based on their shape, size, and color. Utilizing the MobileNetV2 architecture, this research leverages its efficiency and lightweight nature, making it suitable for mobile and embedded applications. Key techniques such as depthwise separable convolutions, linear bottlenecks, and inverted residuals help reduce the number of parameters and computational load while maintaining high performance in feature extraction. Additionally, the study employs comprehensive data augmentation methods, including horizontal and vertical flips, grayscale transformations, hue adjustments, brightness adjustments, and noise addition to enhance the model's robustness and generalization capabilities. The proposed model demonstrates superior performance, achieving an overall accuracy of 99.53%similar to 100% with nearly perfect precision, recall of 95.7%, and F1-score of 94.6% for both "orange_good" and "orange_bad" classes, significantly outperforming previous models which typically achieved accuracies between 70% and 90%. While the classification performance was near-perfect in some aspects, there were minor errors in specific detection tasks. The confusion matrix shows that the model has high sensitivity and specificity, with very few misclassifications. Finally, this study highlights the practical applicability of the proposed model, particularly its easy deployment on resource-constrained devices and its effectiveness in agricultural product quality control processes. These findings affirm the model in this research as a reliable and highly efficient tool for agricultural product classification, surpassing the capabilities of traditional models in this field.
引用
收藏
页数:22
相关论文
共 50 条
  • [1] Automated Visual Quality Detection for Tilapia using MobilenetV2 Convolutional Neural Network
    Breta, Israel F.
    Catalan, Karl Adriane D. C.
    Constantino, Sev Kristian M.
    Mones, Ralph Adrian R.
    Bustillos, Edward D.
    2024 IEEE INTERNATIONAL CONFERENCE ON SMART COMPUTING, SMARTCOMP 2024, 2024, : 296 - 301
  • [2] Convolutional Neural Network Implementation with MobileNetV2 Architecture for Indonesian Herbal Plants Classification in Mobile App
    Kusumaningtyas, Entin Martiana
    Ramadijanti, Nana
    Rijal, Irfan Hanif Khoiru
    2024 INTERNATIONAL ELECTRONICS SYMPOSIUM, IES 2024, 2024, : 521 - 527
  • [3] Data augmentation based morphological classification of galaxies using deep convolutional neural network
    Ansh Mittal
    Anu Soorya
    Preeti Nagrath
    D. Jude Hemanth
    Earth Science Informatics, 2020, 13 : 601 - 617
  • [4] Environmental sound classification using a regularized deep convolutional neural network with data augmentation
    Mushtaq, Zohaib
    Su, Shun-Feng
    APPLIED ACOUSTICS, 2020, 167
  • [5] Data augmentation based morphological classification of galaxies using deep convolutional neural network
    Mittal, Ansh
    Soorya, Anu
    Nagrath, Preeti
    Hemanth, D. Jude
    EARTH SCIENCE INFORMATICS, 2020, 13 (03) : 601 - 617
  • [6] FACEMASK DETECTION USING CONVOLUTIONAL NEURAL NETWORKS (CNN) WITH MOBILENETV2 ARCHITECTURE
    Jubilson, Ajith
    Madhurya, Ch
    INTERNATIONAL JOURNAL OF EARLY CHILDHOOD SPECIAL EDUCATION, 2022, 14 (01) : 2173 - 2183
  • [7] Sheep face recognition and classification based on an improved MobilenetV2 neural network
    Pang, Yue
    Yu, Wenbo
    Zhang, Yongan
    Xuan, Chuanzhong
    Wu, Pei
    INTERNATIONAL JOURNAL OF ADVANCED ROBOTIC SYSTEMS, 2023, 20 (01)
  • [8] Avocado fruit disease detection and classification using modified SCA–PSO algorithm-based MobileNetV2 convolutional neural network
    Satyasis Mishra
    Tadesse Hailu Ayane
    V. Ellappan
    Davinder Singh Rathee
    Harish Kalla
    Iran Journal of Computer Science, 2022, 5 (4) : 345 - 358
  • [9] Roman Amphitheater Classification Using Convolutional Neural Network and Data Augmentation
    Nakouri, Haifa
    PARALLEL AND DISTRIBUTED COMPUTING, APPLICATIONS AND TECHNOLOGIES, PDCAT 2021, 2022, 13148 : 476 - 484
  • [10] Data augmentation techniques for transfer learning improvement in drill wear classification using convolutional neural network
    Kurek, Jaroslaw
    Antoniuk, Izabella
    Górski, Jaroslaw
    Jegorowa, Albina
    Świderski, Bartosz
    Kruk, Michal
    Wieczorek, Grzegorz
    Pach, Jakub
    Orlowski, Arkadiusz
    Aleksiejuk-Gawron, Joanna
    Machine Graphics and Vision, 2019, 28 (1-4): : 3 - 12