Abstract:The number of wheat ears in a field is an important parameter for accurately estimating wheat yield. In a large field, however, it is hard to conduct an automated and accurate counting of wheat ears because of their density and mutual overlay. Unlike the majority of the studies conducted on deep learning-based methods that usually count wheat ears via a collection of static images, this paper proposes a counting method based directly on a UAV video multi-objective tracking method and better counting efficiency results. Firstly, we optimized the YOLOv7 model because the basis of the multi-target tracking algorithm is target detection. Simultaneously, the omni-dimensional dynamic convolution (ODConv) design was applied to the network structure to significantly improve the feature-extraction capability of the model, strengthen the interaction between dimensions, and improve the performance of the detection model. Furthermore, the global context network (GCNet) and coordinate attention (CA) mechanisms were adopted in the backbone network to implement the effective utilization of wheat features. Secondly, this study improved the DeepSort multi-objective tracking algorithm by replacing the DeepSort feature extractor with a modified ResNet network structure to achieve a better extraction of wheat-ear-feature information, and the constructed dataset was then trained for the re-identification of wheat ears. Finally, the improved DeepSort algorithm was used to calculate the number of different IDs that appear in the video, and an improved method based on YOLOv7 and DeepSort algorithms was then created to calculate the number of wheat ears in large fields. The results show that the mean average precision (mAP) of the improved YOLOv7 detection model is 2.5% higher than that of the original YOLOv7 model, reaching 96.2%. The multiple-object tracking accuracy (MOTA) of the improved YOLOv7–DeepSort model reached 75.4%. By verifying the number of wheat ears captured by the UAV method, it can be determined that the average value of an L1 loss is 4.2 and the accuracy rate is between 95 and 98%; thus, detection and tracking methods can be effectively performed, and the efficient counting of wheat ears can be achieved according to the ID value in the video.

A method for detecting the number of wheat ears based on the improved YOLOX model

Improved YOLO v5 Wheat Ear Detection Algorithm Based on Attention Mechanism

Research on the Method of Counting Wheat Ears via Video Based on Improved YOLOv7 and DeepSort

YOLOv7-MA: Improved YOLOv7-Based Wheat Head Detection and Counting

A lightweight network for improving wheat ears detection and counting based on YOLOv5s

Detection method of wheat spike improved YOLOv5s based on the attention mechanism

Detection and counting of wheat ear using YOLOv8

Application of Efficient Channel Attention and Small-Scale Layer to YOLOv5s for Wheat Ears Detection

Real-Time Detection and Counting of Wheat Spikes Based on Improved YOLOv10

Occlusion Robust Wheat Ear Counting Algorithm Based on Deep Learning

Winter wheat ear counting based on improved YOLOv7x and Kalman filter tracking algorithm with video streaming

Lightweight YOLOv8 for Wheat Head Detection

Wheat Seed Detection and Counting Method Based on Improved YOLOv8 Model

Improved YOLO-FastestV2 wheat spike detection model based on a multi-stage attention mechanism with a LightFPN detection head

Wheat ear detection based on FasterCANet-YOLOv8s algorithm

A density map-based method for counting wheat ears

IOCSegFormer: Enhancing Wheat Ears Counting in Field Conditions Through Augmented Local Features

Real-time determination of flowering period for field wheat based on improved YOLOv5s model

YOLO-SDL: a lightweight wheat grain detection technology based on an improved YOLOv8n model

Wheat ear counting using K-means clustering segmentation and convolutional neural network