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.

Soybean Seed Counting Based on Pod Image Using Two-Column Convolution Neural Network

High-Throughput Classification and Counting of Vegetable Soybean Pods Based on Deep Learning

Accurate and fast implementation of soybean pod counting and localization from high-resolution image

SoybeanNet: Transformer-Based Convolutional Neural Network for Soybean Pod Counting from Unmanned Aerial Vehicle (UAV) Images

High-throughput soybean seeds phenotyping with convolutional neural networks and transfer learning

An Efficient and Low-Cost Deep Learning-Based Method for Counting and Sizing Soybean Nodules

DeepCorn: A semi-supervised deep learning method for high-throughput image-based corn kernel counting and yield estimation

Segmentation and Phenotype Calculation of Rapeseed Pods Based on YOLO v8 and Mask R-Convolution Neural Networks

Soybean Yield Preharvest Prediction Based on Bean Pods and Leaves Image Recognition Using Deep Learning Neural Network Combined With GRNN

High-throughput soybean pods high-quality segmentation and seed-per-pod estimation for soybean plant breeding

A synthetic datasets based instance segmentation network for High-throughput soybean pods phenotype investigation

Recognition of soybean pods and yield prediction based on improved deep learning model

High-Precision Automated Soybean Phenotypic Feature Extraction Based on Deep Learning and Computer Vision

DeepSeedling: deep convolutional network and Kalman filter for plant seedling detection and counting in the field

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

Estimation of soybean yield based on high-throughput phenotyping and machine learning

Robust soybean seed yield estimation using high-throughput ground robot videos

Rapidly and exactly determining postharvest dry soybean seed quality based on machine vision technology.

Automation of leaf counting in maize and sorghum using deep learning

Automatic kernel counting on maize ear using RGB images

Automatic Detection and Counting of Wheat Spikelet Using Semi-Automatic Labeling and Deep Learning