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

Plant Methods

Shuaijie Shen¹  Fei Yin¹  Xinming Ma¹  Binchao Jiang¹  Hongbo Qiao¹  Xin Xu¹  Zhaowu Ma¹  Haiyang Li¹  Lei Xi¹ 

[1] Henan Agricultural University;

关键词: Wheat ear counting;    Crop yield;    Deep learning;    CNN;    K-means;    Segmentation;   

DOI  :  10.1186/s13007-020-00648-8

来源: DOAJ

【 摘 要 】

Abstract Background Wheat yield is influenced by the number of ears per unit area, and manual counting has traditionally been used to estimate wheat yield. To realize rapid and accurate wheat ear counting, K-means clustering was used for the automatic segmentation of wheat ear images captured by hand-held devices. The segmented data set was constructed by creating four categories of image labels: non-wheat ear, one wheat ear, two wheat ears, and three wheat ears, which was then was sent into the convolution neural network (CNN) model for training and testing to reduce the complexity of the model. Results The recognition accuracy of non-wheat, one wheat, two wheat ears, and three wheat ears were 99.8, 97.5, 98.07, and 98.5%, respectively. The model R 2 reached 0.96, the root mean square error (RMSE) was 10.84 ears, the macro F1-score and micro F1-score both achieved 98.47%, and the best performance was observed during late grain-filling stage (R 2 = 0.99, RMSE = 3.24 ears). The model could also be applied to the UAV platform (R 2 = 0.97, RMSE = 9.47 ears). Conclusions The classification of segmented images as opposed to target recognition not only reduces the workload of manual annotation but also improves significantly the efficiency and accuracy of wheat ear counting, thus meeting the requirements of wheat yield estimation in the field environment.

【 授权许可】

Unknown