Weed and rice seedling detection in field based on MAL-YOLOv5 model
-
Graphical Abstract
-
Abstract
Weeds severely reduce rice yield and quality, making reliable in-field detection of weed and rice seedling essential for automated weed management. Although deep learning-based object detection techniques have shown significant potential in automatically distinguishing crops from weeds, existing models often suffer from large model sizes, high computational complexity, and insufficient feature extraction. To address these issues, this study proposes a lightweight multi-angle object detection model named MAL-YOLOv5 (Multi-Angle Lightweight YOLOv5), based on the YOLOv5 (You Only Look Once version 5) framework, which effectively reduces model complexity while maintaining detection performance. Specifically, the lightweight MobileNetV3 architecture is adopted to replace the original backbone network, significantly decreasing the number of parameters without compromising accuracy. Furthermore, the neck network of MAL-YOLOv5 is enhanced by integrating spatial and channel reconstruction convolution (SCConv) and a single-shot feature aggregation module (SCCSP), which reduces spatial and channel redundancies in the convolutional module, thereby compressing the neck network and improving feature representation. Additionally, a rotated bounding box with angular information is introduced for annotating and detecting rice seedling and weed, which effectively mitigates the interference from background and non-target objects, enabling more precise identification. Experimental results show that the precision, recall, and mAP of the MAL-YOLOv5 model are 93.1%, 91.9%, and 93.4%, respectively. Compared to YOLOv5s_obb, the MAL-YOLOv5 model reduces the number of parameters by 80.1% and computational cost in GFLOPs (Giga Floating-Point Operations) by 81.5%, significantly minimizing model size with only marginal loss in accuracy, conserving computational and storage costs while lowering the hardware requirements for intelligent mechanical weeding equipment.
-
-