Abstract:Background: Grain count is crucial to wheat yield composition and estimating yield parameters. However, traditional manual counting methods are time-consuming and labor-intensive. This study developed an advanced deep learning technique for the segmentation counting model of wheat grains. This model has been rigorously tested on three distinct wheat varieties: 'Bainong 307', 'Xinmai 26', and 'Jimai 336', and it has achieved unprecedented predictive counting accuracy. Method: The images of wheat ears were taken with a smartphone at the late stage of wheat grain filling. We used image processing technology to preprocess and normalize the images to 480*480 pixels. A CBAM-HRNet wheat grain segmentation counting deep learning model based on the Convolutional Block Attention Module (CBAM) was constructed by combining deep learning, migration learning, and attention mechanism. Image processing algorithms and wheat grain texture features were used to build a grain counting and predictive counting model for wheat grains. Results: The CBAM-HRNet model using the CBAM was the best for wheat grain segmentation. Its segmentation accuracy of 92.04%, the mean Intersection over Union (mIoU) of 85.21%, the category mean pixel accuracy (mPA) of 91.16%, and the recall rate of 91.16% demonstrate superior robustness compared to other models such as HRNet, PSPNet, DeeplabV3+ , and U-Net. Method I for spike count, which calculates twice the number of grains on one side of the spike to determine the total number of grains, demonstrates a coefficient of determination R2 of 0.85, a mean absolute error (MAE) of 1.53, and a mean relative error (MRE) of 2.91. In contrast, Method II for spike count involves summing the number of grains on both sides to determine the total number of grains, demonstrating a coefficient of determination R2 of 0.92, an MAE) of 1.15, and an MRE) of 2.09%. Conclusions: Image segmentation algorithm of the CBAM-HRNet wheat spike grain is a powerful solution that uses the CBAM to segment wheat spike grains and obtain richer semantic information. This model can effectively address the challenges of small target image segmentation and under-fitting problems in training. Additionally, the spike grain counting model can quickly and accurately predict the grain count of wheat, providing algorithmic support for efficient and intelligent wheat yield estimation.

An image analysis technique for wheat head count detection using machine learning

A Method for Wheat Head Detection Based on Yolov4

WheatNet: A lightweight convolutional neural network for high-throughput image-based wheat head detection and counting

DeepCount: In-Field Automatic Quantification of Wheat Spikes Using Simple Linear Iterative Clustering and Deep Convolutional Neural Networks

Detection and enumeration of wheat grains based on a deep learning method under various scenarios and scales

Occlusion Robust Wheat Ear Counting Algorithm Based on Deep Learning

Real-Time Detection for Wheat Head Applying Deep Neural Network.

Wheat Heads Detection Via Yolov5 with Weighted Coordinate Attention

Comparative Analysis of Deep Neural Network and Artificial Neural Network for Predicting Wheat Water Content Using UAV Multispectral and Thermal Imagers

The Field Wheat Count Based on the Efficientdet Algorithm

Advancing precision agriculture: The potential of deep learning for cereal plant head detection

WheatLFANet: in-field detection and counting of wheat heads with high-real-time global regression network

An Improved Strategy of Wheat Kernel Recognition Based on Deep Learning

Identification Method of Wheat Cultivars by Using a Convolutional Neural Network Combined with Images of Multiple Growth Periods of Wheat

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

Detection of Fusarium Head Blight in Wheat Using a Deep Neural Network and Color Imaging

An automatic detection method to the field wheat based on image processing

Wheat ear detection based on FasterCANet-YOLOv8s algorithm

Wheat Head Counting by Estimating a Density Map with Convolutional Neural Networks

Detection of Wheat Unsound Kernels Based on Improved ResNet

Segmentation and counting of wheat spike grains based on deep learning and textural feature