Abstract:Introduction: The precise detection of weeds in the field is the premise of implementing weed management. However, the similar color, morphology, and occlusion between wheat and weeds pose a challenge to the detection of weeds. In this study, a CSCW-YOLOv7 based on an improved YOLOv7 architecture was proposed to identify five types of weeds in complex wheat fields. Methods: First, a dataset was constructed for five weeds that are commonly found, namely, Descurainia sophia , thistle, golden saxifrage, shepherd's purse herb, and Artemisia argyi . Second, a wheat weed detection model called CSCW-YOLOv7 was proposed to achieve the accurate identification and classification of wheat weeds. In the CSCW-YOLOv7, the CARAFE operator was introduced as an up-sampling algorithm to improve the recognition of small targets. Then, the Squeeze-and-Excitation (SE) network was added to the Extended Latent Attention Networks (ELAN) module in the backbone network and the concatenation layer in the feature fusion module to enhance important weed features and suppress irrelevant features. In addition, the contextual transformer (CoT) module, a transformer-based architectural design, was used to capture global information and enhance self-attention by mining contextual information between neighboring keys. Finally, the Wise Intersection over Union (WIoU) loss function introducing a dynamic nonmonotonic focusing mechanism was employed to better predict the bounding boxes of the occluded weed. Results and discussion: The ablation experiment results showed that the CSCW-YOLOv7 achieved the best performance among the other models. The accuracy, recall, and mean average precision (mAP) values of the CSCW-YOLOv7 were 97.7%, 98%, and 94.4%, respectively. Compared with the baseline YOLOv7, the improved CSCW-YOLOv7 obtained precision, recall, and mAP increases of 1.8%, 1%, and 2.1%, respectively. Meanwhile, the parameters were compressed by 10.7% with a 3.8-MB reduction, resulting in a 10% decrease in floating-point operations per second (FLOPs). The Gradient-weighted Class Activation Mapping (Grad-CAM) visualization method suggested that the CSCW-YOLOv7 can learn a more representative set of features that can help better locate the weeds of different scales in complex field environments. In addition, the performance of the CSCW-YOLOv7 was compared to the widely used deep learning models, and results indicated that the CSCW-YOLOv7 exhibits a better ability to distinguish the overlapped weeds and small-scale weeds. The overall results suggest that the CSCW-YOLOv7 is a promising tool for the detection of weeds and has great potential for field applications.

Application of Efficient Channel Attention and Small-Scale Layer to YOLOv5s for Wheat Ears Detection

Improved YOLO v5 Wheat Ear Detection Algorithm Based on Attention Mechanism

A lightweight network for improving wheat ears detection and counting based on YOLOv5s

Lightweight YOLOv8 for Wheat Head Detection

Wheat ear detection based on FasterCANet-YOLOv8s algorithm

Detection method of wheat spike improved YOLOv5s based on the attention mechanism

Detection and counting of wheat ear using YOLOv8

YOLOv7-MA: Improved YOLOv7-Based Wheat Head Detection and Counting

Real-Time Detection and Counting of Wheat Spikes Based on Improved YOLOv10

Improved YOLO-FastestV2 wheat spike detection model based on a multi-stage attention mechanism with a LightFPN detection head

UAV remote sensing detection and target recognition based on SCP-YOLO

YOLO-SDL: a lightweight wheat grain detection technology based on an improved YOLOv8n model

Weed detection and recognition in complex wheat fields based on an improved YOLOv7

A method for detecting the number of wheat ears based on the improved YOLOX model

Wheat Seed Detection and Counting Method Based on Improved YOLOv8 Model

Efficient Optimized YOLOv8 Model with Extended Vision

YOLOv8 Model for Weed Detection in Wheat Fields Based on a Visual Converter and Multi-Scale Feature Fusion

A Wheat Spike Detection Method in UAV Images Based on Improved YOLOv5

Research on the Method of Counting Wheat Ears via Video Based on Improved YOLOv7 and DeepSort

Real-Time Detection for Wheat Head Applying Deep Neural Network.