GE-YOLO for Weed Detection in Rice Paddy Fields

Weeds are a significant adverse factor affecting rice growth, and their efficient removal necessitates an accurate, efficient, and well-generalizing weed detection method. However, weed detection faces challenges such as a complex vegetation environment, the similar morphology and color of weeds, and crops and varying lighting conditions. The current research has yet to address these issues adequately. Therefore, we propose GE-YOLO to identify three common types of weeds in rice fields in the Hunan province of China and to validate its generalization performance. GE-YOLO is an improvement based on the YOLOv8 baseline model. It introduces the Neck network with the Gold-YOLO feature aggregation and distribution network to enhance the network’s ability to fuse multi-scale features and detect weeds of different sizes. Additionally, an EMA attention mechanism is used to better learn weed feature representations, while a GIOU loss function provides smoother gradients and reduces computational complexity. Multiple experiments demonstrate that GE-YOLO achieves 93.1% mAP, 90.3% F1 Score, and 85.9 FPS, surpassing almost all mainstream object detection algorithms such as YOLOv8, YOLOv10, and YOLOv11 in terms of detection accuracy and overall performance. Furthermore, the detection results under different lighting conditions consistently maintained a high level above 90% mAP, and under conditions of heavy occlusion, the average mAP for all weed types reached 88.7%. These results indicate that GE-YOLO has excellent detection accuracy and generalization performance, highlighting the potential of GE-YOLO as a valuable tool for enhancing weed management practices in rice cultivation.