IDEAS home Printed from https://ideas.repec.org/a/gam/jagris/v14y2024i10p1798-d1497600.html
   My bibliography  Save this article

Object Detection Algorithm for Citrus Fruits Based on Improved YOLOv5 Model

Author

Listed:
  • Yao Yu

    (Guizhou Mountainous Agriculture Machinery Research Institute, Guiyang 550000, China)

  • Yucheng Liu

    (College of Engineering and Technology, Southwest University, Chongqing 400715, China)

  • Yuanjiang Li

    (College of Engineering and Technology, Southwest University, Chongqing 400715, China)

  • Changsu Xu

    (College of Engineering and Technology, Southwest University, Chongqing 400715, China)

  • Yunwu Li

    (College of Engineering and Technology, Southwest University, Chongqing 400715, China)

Abstract

To address the challenges of missed and false detections in citrus fruit detection caused by environmental factors such as leaf occlusion, fruit overlap, and variations in natural light in hilly and mountainous orchards, this paper proposes a citrus detection model based on an improved YOLOv5 algorithm. By introducing receptive field convolutions with full 3D weights (RFCF), the model overcomes the issue of parameter sharing in convolution operations, enhancing detection accuracy. A focused linear attention (FLA) module is incorporated to improve the expressive power of the self-attention mechanism while maintaining computational efficiency. Additionally, anchor boxes were re-clustered based on the shape characteristics of target objects, and the boundary box loss function was improved to Foal-EIoU, boosting the model’s localization ability. Experiments conducted on a citrus fruit dataset labeled using LabelImg, collected from hilly and mountainous areas, showed a detection precision of 95.83% and a mean average precision (mAP) of 79.68%. This research not only significantly improves detection performance in complex environments but also provides crucial data support for precision tasks such as orchard localization and intelligent picking, demonstrating strong potential for practical applications in smart agriculture.

Suggested Citation

  • Yao Yu & Yucheng Liu & Yuanjiang Li & Changsu Xu & Yunwu Li, 2024. "Object Detection Algorithm for Citrus Fruits Based on Improved YOLOv5 Model," Agriculture, MDPI, vol. 14(10), pages 1-25, October.
  • Handle: RePEc:gam:jagris:v:14:y:2024:i:10:p:1798-:d:1497600
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2077-0472/14/10/1798/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2077-0472/14/10/1798/
    Download Restriction: no
    ---><---

    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:jagris:v:14:y:2024:i:10:p:1798-:d:1497600. 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.

    We have no bibliographic references for this item. You can help adding them by using 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.