IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v11y2023i16p3615-d1221650.html
   My bibliography  Save this article

ATC-YOLOv5: Fruit Appearance Quality Classification Algorithm Based on the Improved YOLOv5 Model for Passion Fruits

Author

Listed:
  • Changhong Liu

    (School of Mechanical and Electrical Engineering, Guangzhou University, Guangzhou 510006, China)

  • Weiren Lin

    (School of Mechanical and Electrical Engineering, Guangzhou University, Guangzhou 510006, China)

  • Yifeng Feng

    (School of Mechanical and Electrical Engineering, Guangzhou University, Guangzhou 510006, China)

  • Ziqing Guo

    (School of Electronics and Communication Engineering, Guangzhou University, Guangzhou 510006, China)

  • Zewen Xie

    (School of Physics and Material Science, Guangzhou University, Guangzhou 510006, China)

Abstract

Passion fruit, renowned for its significant nutritional, medicinal, and economic value, is extensively cultivated in subtropical regions such as China, India, and Vietnam. In the production and processing industry, the quality grading of passion fruit plays a crucial role in the supply chain. However, the current process relies heavily on manual labor, resulting in inefficiency and high costs, which reflects the importance of expanding the application of fruit appearance quality classification mechanisms based on computer vision. Moreover, the existing passion fruit detection algorithms mainly focus on real-time detection and overlook the quality-classification aspect. This paper proposes the ATC-YOLOv5 model based on deep learning for passion fruit detection and quality classification. First, an improved Asymptotic Feature Pyramid Network (APFN) is utilized as the feature-extraction network, which is the network modified in this study by adding weighted feature concat pathways. This optimization enhances the feature flow between different levels and nodes, allowing for the adaptive and asymptotic fusion of richer feature information related to passion fruit quality. Secondly, the Transformer Cross Stage Partial (TRCSP) layer is constructed based on the introduction of the Multi-Head Self-Attention (MHSA) layer in the Cross Stage Partial (CSP) layer, enabling the network to achieve a better performance in modeling long-range dependencies. In addition, the Coordinate Attention (CA) mechanism is introduced to enhance the network’s learning capacity for both local and non-local information, as well as the fine-grained features of passion fruit. Moreover, to validate the performance of the proposed model, a self-made passion fruit dataset is constructed to classify passion fruit into four quality grades. The original YOLOv5 serves as the baseline model. According to the experimental results, the mean average precision (mAP) of ATC-YOLOv5 reaches 95.36%, and the mean detection time (mDT) is 3.2 ms, which improves the mAP by 4.83% and the detection speed by 11.1%, and the number of parameters is reduced by 10.54% compared to the baseline, maintaining the lightweight characteristics while improving the accuracy. These experimental results validate the high detection efficiency of the proposed model for fruit quality classification, contributing to the realization of intelligent agriculture and fruit industries.

Suggested Citation

  • Changhong Liu & Weiren Lin & Yifeng Feng & Ziqing Guo & Zewen Xie, 2023. "ATC-YOLOv5: Fruit Appearance Quality Classification Algorithm Based on the Improved YOLOv5 Model for Passion Fruits," Mathematics, MDPI, vol. 11(16), pages 1-20, August.
  • Handle: RePEc:gam:jmathe:v:11:y:2023:i:16:p:3615-:d:1221650
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/11/16/3615/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/11/16/3615/
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Kathiresan Shankar & Sachin Kumar & Ashit Kumar Dutta & Ahmed Alkhayyat & Anwar Ja’afar Mohamad Jawad & Ali Hashim Abbas & Yousif K. Yousif, 2022. "An Automated Hyperparameter Tuning Recurrent Neural Network Model for Fruit Classification," Mathematics, MDPI, vol. 10(13), pages 1-18, July.
    2. Yu-Huei Cheng & Cheng-Yen Tseng & Duc-Man Nguyen & Yu-Da Lin, 2022. "YOLOv4-Driven Appearance Grading Filing Mechanism: Toward a High-Accuracy Tomato Grading Model through a Deep-Learning Framework," Mathematics, MDPI, vol. 10(18), pages 1-12, September.
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Nannan Xu & Xinze Cui & Xin Wang & Wei Zhang & Tianyu Zhao, 2022. "An Intelligent Athlete Signal Processing Methodology for Balance Control Ability Assessment with Multi-Headed Self-Attention Mechanism," Mathematics, MDPI, vol. 10(15), pages 1-16, August.
    2. Ruslan Abdulkadirov & Pavel Lyakhov & Nikolay Nagornov, 2023. "Survey of Optimization Algorithms in Modern Neural Networks," Mathematics, MDPI, vol. 11(11), pages 1-37, May.

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:11:y:2023:i:16:p:3615-:d:1221650. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.