IDEAS home Printed from https://ideas.repec.org/a/plo/pone00/0295565.html
   My bibliography  Save this article

Sugarcane stem node identification algorithm based on improved YOLOv5

Author

Listed:
  • Zhongjian Xie
  • Yuanhang Li
  • Yao Xiao
  • Yinzhou Diao
  • Hengyu Liao
  • Yaya Zhang
  • Xinwei Chen
  • Weilin Wu
  • Chunming Wen
  • Shangping Li

Abstract

Identification of sugarcane stem nodes is generally dependent on high-performance recognition equipment in sugarcane seed pre-cutting machines and inefficient. Accordingly, this study proposes a novel lightweight architecture for the detection of sugarcane stem nodes based on the YOLOv5 framework, named G-YOLOv5s-SS. Firstly, the study removes the CBS and C3 structures at the end of the backbone network to fully utilize shallow-level feature information. This enhances the detection performance of sugarcane stem nodes. Simultaneously, it eliminates the 32 times down-sampled branches in the neck structure and the 20x20 detection heads at the prediction end, reducing model complexity. Secondly, a Ghost lightweight module is introduced to replace the conventional convolution module in the BottleNeck structure, further reducing the model’s complexity. Finally, the study incorporates the SimAM attention mechanism to enhance the extraction of sugarcane stem node features without introducing additional parameters. This improvement aims to enhance recognition accuracy, compensating for any loss in precision due to lightweight modifications. The experimental results showed that the average precision of the improved network for sugarcane stem node identification reached 97.6%, which was 0.6% higher than that of the YOLOv5 baseline network. Meanwhile, a model size of 2.6MB, 1,129,340 parameters, and 7.2G FLOPs, representing respective reductions of 82%, 84%, and 54.4%. Compared with mainstream one-stage target detection algorithms such as YOLOv4-tiny, YOLOv4, YOLOv5n, YOLOv6n, YOLOv6s, YOLOv7-tiny, and YOLOv7, G-YOLOv5s-SS achieved respective average precision improvements of 12.9%, 5.07%, 3.6%, 2.1%, 1.2%, 3%, and 0.4% in sugarcane stem nodes recognition. Meanwhile, the model size was compressed by 88.9%, 98.9%, 33.3%, 72%, 92.9%, 78.8% and 96.3%, respectively. Compared with similar studies, G-YOLOv5s-SS not only enhanced recognition accuracy but also considered model size, demonstrating an overall excellent performance that aligns with the requirements of sugarcane seed pre-cutting machines.

Suggested Citation

  • Zhongjian Xie & Yuanhang Li & Yao Xiao & Yinzhou Diao & Hengyu Liao & Yaya Zhang & Xinwei Chen & Weilin Wu & Chunming Wen & Shangping Li, 2023. "Sugarcane stem node identification algorithm based on improved YOLOv5," PLOS ONE, Public Library of Science, vol. 18(12), pages 1-21, December.
  • Handle: RePEc:plo:pone00:0295565
    DOI: 10.1371/journal.pone.0295565
    as

    Download full text from publisher

    File URL: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0295565
    Download Restriction: no

    File URL: https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0295565&type=printable
    Download Restriction: no

    File URL: https://libkey.io/10.1371/journal.pone.0295565?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    References listed on IDEAS

    as
    1. Yuanzhou Zheng & Yuanfeng Zhang & Long Qian & Xinzhu Zhang & Shitong Diao & Xinyu Liu & Jingxin Cao & Haichao Huang, 2023. "A lightweight ship target detection model based on improved YOLOv5s algorithm," PLOS ONE, Public Library of Science, vol. 18(4), pages 1-23, April.
    2. Hassaan Malik & Ahmad Naeem & Shahzad Hassan & Farman Ali & Rizwan Ali Naqvi & Dong Keon Yon, 2023. "Multi-classification deep neural networks for identification of fish species using camera captured images," PLOS ONE, Public Library of Science, vol. 18(4), pages 1-32, April.
    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. Yuping Yin & Zheyu Zhang & Lin Wei & Chao Geng & Haoxiang Ran & Haodong Zhu, 2023. "Pedestrian detection algorithm integrating large kernel attention and YOLOV5 lightweight model," PLOS ONE, Public Library of Science, vol. 18(11), pages 1-23, November.

    More about this item

    Statistics

    Access and download statistics

    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:plo:pone00:0295565. 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: plosone (email available below). General contact details of provider: https://journals.plos.org/plosone/ .

    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.