Tea Buds Detection in Complex Background Based on Improved YOLOv7

Abstract

Aiming at the problem that the color of tea buds is highly similar to the background in complex scenes and it is difficult to identify the buds, this study proposed an improved YOLOv7 algorithm by replacing the original convolution blocks with Depth Separable Convolution (DS Conv) blocks, and adding Convolutional Block Attention Modules (CBAM) and Coordinate Attention (CA) modules. The method improved mean Average Precision (mAP) by 1.28% and mean Recall (mR) rate by 2.92%, the final mAP and mR reached 96.70% and 93.88%, respectively, and 30.62 Frame Per Second (FPS) of the improved model meets the requirements of real-time detection. The results show that the detection accuracy of the improved YOLOv7 algorithm for tea buds was higher than that of other target detection algorithms, and the detecting performance is not significantly affected by the light conditions, and the recognition accuracy of tea buds at each growing period was excellent and balanced. This study provides experience for the realization of intelligent tea picking.

Keywords

• Attention module
• deep learning
• tea buds
• target detection
• YOLOv7