Abstract <p>Addressing issues such as high parameter count and low detection accuracy in weed detection algorithms, this paper presents a study on a weed detection algorithm based on an improved YOLOv5s model. Firstly, the backbone network was enhanced using the ParC modul4e to reduce computational demands and increase model detection speed; secondly, the C3BRA module, designed based on the BRA attention mechanism, replaced the original C3 module to focus on the extraction and reinforcement of key feature information; finally, the SIoU loss function replaced the CIoU loss function, accelerating network convergence and improving model detection accuracy. Experimental validation on the test dataset compared the improved model with the original YOLOv5s model, showing that the modified model increased the <i>P</i> value by 2.8%, the mAP value by 1.7%, and reduced model parameters by 10.7%, better meeting the requirements for weed detection.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Weed Detection Algorithm Based on Improved YOLOv5S

  • Yuehua Li,
  • Zhangyan Yao,
  • Yuyang Gu,
  • Bin Hu

摘要

Abstract

Addressing issues such as high parameter count and low detection accuracy in weed detection algorithms, this paper presents a study on a weed detection algorithm based on an improved YOLOv5s model. Firstly, the backbone network was enhanced using the ParC modul4e to reduce computational demands and increase model detection speed; secondly, the C3BRA module, designed based on the BRA attention mechanism, replaced the original C3 module to focus on the extraction and reinforcement of key feature information; finally, the SIoU loss function replaced the CIoU loss function, accelerating network convergence and improving model detection accuracy. Experimental validation on the test dataset compared the improved model with the original YOLOv5s model, showing that the modified model increased the P value by 2.8%, the mAP value by 1.7%, and reduced model parameters by 10.7%, better meeting the requirements for weed detection.