An Improved YOLOv5-Based Tapping Trajectory Detection Method for Natural Rubber Trees

摘要

The object detection algorithm is one of the core technologies of the intelligent rubber tapping robot, but most of the existing detection algorithms cannot effectively meet the tapping trajectory detection of natural rubber trees in the complex forest environment. This paper proposes a tapping trajectory detection method for natural rubber trees based on an improved YOLOv5 model to accomplish fast and accurate detection. Firstly, the coordinate attention (CA) mechanism is added to the Backbone network to embed the location information into the channel attention, which effectively improves the detection accuracy. Secondly, a module called convolution and GhostBottleneck (CGB) is designed, based on the Ghost module, to substitute the Cross Stage Partial Network (CSP) module in the Neck network, which ensures the detection accuracy while reducing model parameters. Finally, the EIoU loss function is introduced to enable a more accurate regression of the model. The experimental results show that the overall performance of the YOLOv5-CCE model outperforms the original YOLOv5 and other classical lightweight detection algorithms. Compared with the original YOLOv5 model, the YOLOv5-CCE model has a 2.1% improvement in mAP value, a 2.5% compression of model parameters, and a 7.0% reduction in the number of floating point operations (FLOPs). Therefore, the improved model can fully meet the requirements of real-time detection, providing a robust detection method for rubber tapping robots.