Wheat Heads Detection Via Yolov5 with Weighted Coordinate Attention

Feixiang Han,Jie Li
DOI: https://doi.org/10.1109/icccbda55098.2022.9778925
2022-01-01
Abstract:Wheat is one of the three crops globally. Accurately detecting and counting wheat heads is essential for crop yield assessment and breeding. The traditional method involves manually counting the number of wheat heads, which is time-consuming and inefficient. Although researchers have studied several algorithms to measure the wheat head automatically, the generalizability limits the application of these algorithms. Our research aims to develop a method with high accuracy and generalizability for wheat head detection in the field. The proposed method employed YOLOv5, WCA(Weighted Coordinate Attention), and image-processing techniques to detect the wheat head. WCA uses the weighted average to aggregate the feature information along with the x and y directions, capturing the long-range dependencies among the feature maps. Via WCA, the Yolov5 could extract comprehensive feature information of the wheat head in complex environments, improving the detection performance of the model. Benchmarking Yolov5-WCA on 6500 labeled images of the wheat head, we achieved the average precision, recall, accuracy of wheat head detection of 93.7%, 91.6%, 86.2%, outperforms Yolov5 by 0.7% on precision, 1.9% on recall, and 2.2% on accuracy. The proposed method can extract comprehensive feature information among complex environments, improve the model's accuracy and generalizability, and provide technical support for other detection tasks.
What problem does this paper attempt to address?