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.

Estimating the Size of Densely Packed and Complex Grains Using Valley Edge Detection with a One-Pixel Width and Grain Shape Information

A novel segmentation algorithm for clustered flexional agricultural products based on image analysis.

Novel Segmentation Method and Measurement System for Various Grains with Complex Touching.

CNN for Image-Based Sediment Detection Applied to a Large Terrestrial and Airborne Dataset

Polyamine cytotoxicity in the presence of bovine serum amine oxidase.

Grape Berry Detection and Size Measurement Based on Edge Image Processing and Geometric Morphology

Cereal grain 3D point cloud analysis method for shape extraction and filled/unfilled grain identification based on structured light imaging

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

Convolutional Neural Networks for Image-Based Sediment Detection Applied to a Large Terrestrial and Airborne Dataset

Embedded Yolo-Fastest V2-Based 3D Reconstruction and Size Prediction of Grain Silo-Bag

Application of Image Segmentation Algorithm Based on Entropy Clustering in Apple Harvesting Robot

Research on a rapid identification method for counting universal grain crops

Size Measurement of Stone Images Based on Improved UNet

Image-Based 3D Reconstruction of Granular Grains via Hybrid Algorithm and Level Set with Convolution Kernel

An annotated grain kernel image database for visual quality inspection

An automated method for the assessment of the rice grain germination rate

Panicle-SEG: a robust image segmentation method for rice panicles in the field based on deep learning and superpixel optimization

Environmental influences on annual variation in nest success of smallmouth bass,Micropterus dolomieui, in Long Point Bay, Lake Erie

Enhanced Hybrid Algorithms for Segmentation and Reconstruction of Granular Grains From X‐Ray Micro Computed‐Tomography Images

The Improvement of Adhesive Grain Particles Image Segmentation Algorithm Based on the Mathematical Morphology

Crop Row Segmentation and Detection in Paddy Fields Based on Treble-Classification Otsu and Double-Dimensional Clustering Method