Lightweight highland barley detection based on improved YOLOv5
摘要
Accurate and efficient assessment of highland barley (Hordeum vulgare L.) density is crucial for optimizing cultivation and management practices. However, challenges such as overlapping spikes in unmanned aerial vehicle (UAV) images and the computational requirements for high-resolution image analysis hinder real-time detection capabilities. To address these issues, this study proposes an improved lightweight YOLOv5 model for highland barley spike detection. We chose depthwise separable convolution (DSConv) and ghost convolution (GhostConv) for the backbone and neck networks, respectively, to reduce the parameter and computational complexity. In addition, the integration of convolutional block attention module (CBAM) enhances the model’s ability to focus on target object in complex backgrounds. The results show that the improved YOLOv5 model has a significant improvement in detection performance. Precision and recall increased by 3.1% to 92.2% and 86.2%, respectively, with an F1 score of 0.892. The