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

High-Precision Detection for Sandalwood Trees via Improved YOLOv5s and StyleGAN

Author

Listed:
  • Yu Zhang

    (College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University, Guangzhou 510642, China)

  • Jiajun Niu

    (College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University, Guangzhou 510642, China)

  • Zezhong Huang

    (College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University, Guangzhou 510642, China)

  • Chunlei Pan

    (College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University, Guangzhou 510642, China)

  • Yueju Xue

    (College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University, Guangzhou 510642, China)

  • Fengxiao Tan

    (College of Natural Resources and Environment, South China Agricultural University, Guangzhou 510642, China)

Abstract

An algorithm model based on computer vision is one of the critical technologies that are imperative for agriculture and forestry planting. In this paper, a vision algorithm model based on StyleGAN and improved YOLOv5s is proposed to detect sandalwood trees from unmanned aerial vehicle remote sensing data, and this model has excellent adaptability to complex environments. To enhance feature expression ability, a CA (coordinate attention) module with dimensional information is introduced, which can both capture target channel information and keep correlation information between long-range pixels. To improve the training speed and test accuracy, SIOU (structural similarity intersection over union) is proposed to replace the traditional loss function, whose direction matching degree between the prediction box and the real box is fully considered. To achieve the generalization ability of the model, StyleGAN is introduced to augment the remote sensing data of sandalwood trees and to improve the sample balance of different flight heights. The experimental results show that the average accuracy of sandalwood tree detection increased from 93% to 95.2% through YOLOv5s model improvement; then, on that basis, the accuracy increased by another 0.4% via data generation from the StyleGAN algorithm model, finally reaching 95.6%. Compared with the mainstream lightweight models YOLOv5-mobilenet, YOLOv5-ghost, YOLOXs, and YOLOv4-tiny, the accuracy of this method is 2.3%, 2.9%, 3.6%, and 6.6% higher, respectively. The size of the training sandalwood tree model is 14.5 Mb, and the detection time is 17.6 ms. Thus, the algorithm demonstrates the advantages of having high detection accuracy, a compact model size, and a rapid processing speed, making it suitable for integration into edge computing devices for on-site real-time monitoring.

Suggested Citation

  • Yu Zhang & Jiajun Niu & Zezhong Huang & Chunlei Pan & Yueju Xue & Fengxiao Tan, 2024. "High-Precision Detection for Sandalwood Trees via Improved YOLOv5s and StyleGAN," Agriculture, MDPI, vol. 14(3), pages 1-13, March.
  • Handle: RePEc:gam:jagris:v:14:y:2024:i:3:p:452-:d:1354754
    as

    Download full text from publisher

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

    File URL: https://www.mdpi.com/2077-0472/14/3/452/
    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:3:p:452-:d:1354754. 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.