STBNA-YOLOv5: An Improved YOLOv5 Network for Weed Detection in Rapeseed Field

Abstract

Rapeseed is one of the primary oil crops; yet, it faces significant threats from weeds. The ideal method for applying herbicides would be selective variable spraying, but the primary challenge lies in automatically identifying weeds. To address the issues of dense weed identification, frequent occlusion, and varying weed sizes in rapeseed fields, this paper introduces a STBNA-YOLOv5 weed detection model and proposes three enhanced algorithms: incorporating a Swin Transformer encoder block to bolster feature extraction capabilities, utilizing a BiFPN structure coupled with a NAM attention mechanism module to efficiently harness feature information, and incorporating an adaptive spatial fusion module to enhance recognition sensitivity. Additionally, the random occlusion technique and weed category image data augmentation method are employed to diversify the dataset. Experimental results demonstrate that the STBNA-YOLOv5 model outperforms detection models such as SDD, Faster-RCNN, YOLOv3, DETR, and EfficientDet in terms of Precision, F1-score, and [email protected], achieving scores of 0.644, 0.825, and 0.908, respectively. For multi-target weed detection, the study presents detection results under various field conditions, including sunny, cloudy, unobstructed, and obstructed. The results indicate that the weed detection model can accurately identify both rapeseed and weed species, demonstrating high stability.

Topics & Concepts