中国农业科学 ›› 2024, Vol. 57 ›› Issue (20): 3974-3985.doi: 10.3864/j.issn.0578-1752.2024.20.003

• 耕作栽培·生理生化·农业信息技术 • 上一篇    下一篇

基于双输出回归卷积神经网络的燕麦株高估测研究

张建龙1,2(), 邢文文1, 叶绍波1,2, 张超1,2, 郑德聪1,2()   

  1. 1 山西农业大学农业工程学院,山西太谷 030801
    2 旱作农业机械关键技术与装备山西省重点实验室,山西太谷 030801
  • 收稿日期:2024-03-13 接受日期:2024-09-19 出版日期:2024-10-16 发布日期:2024-10-24
  • 通信作者:
    郑德聪,E-mail:
  • 联系方式: 张建龙,E-mail:zhjl@sxau.edu.cn。
  • 基金资助:
    中央引导地方科技发展资金项目(YDZJSX20231C009); 山西农业大学博士科研启动项目(2021BQ85); 山西省博士毕业生; 博士后研究人员来晋工作奖励资金科研项目(SXBYKY2022019); 山西农业大学学术恢复项目(2023XSHF2)

Oat Plant Height Estimation Based on a Dual Output Regression Convolutional Neural Network

ZHANG JianLong1,2(), XING WenWen1, YE ShaoBo1,2, ZHANG Chao1,2, ZHENG DeCong1,2()   

  1. 1 College of Agricultural Engineering, Shanxi Agricultural University, Taigu 030801, Shanxi
    2 Dryland Farm Machinery Key Technology and Equipment Key Laboratory of Shanxi Province, Taigu 030801, Shanxi
  • Received:2024-03-13 Accepted:2024-09-19 Published:2024-10-16 Online:2024-10-24

摘要:

【目的】株高影响燕麦的单株生产力,并与种植密度共同作用影响单位面积产量。探索大田环境下燕麦株高参数的自动、实时、精准获取方法,以期为燕麦田间自动化管理提供技术参考。【方法】首先基于Intel RealSense D435型深度相机和LabVIEW软件开发平台搭建燕麦深度图像采集系统,以‘品燕4号’燕麦为研究对象,获取生长全程26 376组建模数据和2 205组测试数据,每幅深度图像中燕麦所对应的平均株高和最高株高使用量尺测得。建模数据和测试数据在燕麦各株高区间内的数量相对均衡,并对图像进行高度还原、灰度化和缩放的简单预处理,随之给每张图像打2张标签,分别为图像中燕麦的平均株高和最高株高。基于8种经典卷积神经网络模型,将各网络模型的最后一层(分类层)去除,添加2个单节点且没有激活函数的全连接层后,分别构建双输出回归卷积神经网络估测模型,模型使用均方差函数(mean square error,MSE)评价各模型估测燕麦株高时的准确率。最终基于TensorFlow深度学习平台,采用建模数据经5折交叉验证选取Modified EfficientNet V2 L为估测模型。【结果】采用未参与模型训练的测试数据考察了Modified EfficientNet V2 L模型估测燕麦株高的泛化性能,该模型估测燕麦平均株高时平均绝对误差(mean absolute error,MAE)、均方根差(root mean square error,RMSE)和平均相对误差(mean relative error,MRE)分别为2.30 cm、2.90 cm和4.4%,估测最高株高时分别为2.24 cm、2.82 cm和4.1%,模型平均估测时间为52.14 ms。使用该方法估测作物株高时的精度与已有方法相近,平均估测时间可以满足作物株高获取的实时性要求。燕麦平均株高和最高株高估测时的相对误差随着作物株高的增加呈总体下降趋势,可能是由于作物株高较低时,估测结果受土壤起伏度影响较大。模型特征图可视化的结果表明,模型根据深度图像中燕麦的高度及轮廓对株高进行估测。最终基于LabVIEW软件开发平台构建了燕麦株高估测系统,系统在获取燕麦深度图像后,可以在0.1 s内精准估测出燕麦平均株高和最高株高,整个过程无需人为干预。【结论】使用深度图像和双输出回归卷积神经网络可以估测燕麦株高,其精度可以满足生产需求,该方法可为燕麦等作物的田间管理提供依据。

关键词: 立体图像, 深度学习, 株高, LabVIEW, 燕麦

Abstract:

【Objective】 Oat plant height affects the productivity per plant and the yield per unit area together with planting density. This study explores automatic, real-time, and precise methods for acquiring oat plant height in a field environment, aiming to provide technical references for the automated field management of oat. 【Method】 Firstly, an oat depth image acquisition system was built based on Intel RealSense D435 depth camera and LabVIEW software development platform. Taking Oat ‘Pinyan No. 4’ as the research object, 26 376 modeling data and 2 205 test data were obtained during the whole oat growth process. The average and highest plant height of oats in each depth image were measured with a scale. The quantity of modeling data and test data in each height range of oat plant was relatively balanced. The images were preprocessed by high restoration, grayscale and scaling. Each image was tagged with two labels, one for the average and one for the highest plant height of the oats in the image. Then, based on 8 classical convolutional neural network models, the last layer (classification layer) of each network model was removed, and two fully connected layers with single nodes and no activation function were added to construct the double output regression convolutional neural network estimation model. Mean square error (MSE) was used to evaluate the accuracy of each model in estimating oat plant height. Finally, based on the TensorFlow deep learning platform, Modified EfficientNet V2L was selected as the estimation model by 5-fold cross-validation using the modeling data. 【Result】 The generalization performance of Modified EfficientNet V2L model to estimate oat plant height was investigated using test data not involved in model training. The mean absolute error (MAE), root mean square error (RMSE) and mean relative error (MRE) to estimate oat average plant height were 2.30 cm, 2.90 cm and 4.4%, respectively. Meanwhile, the MAE, RMSE and MRE to estimate highest plant height was 2.24 cm, 2.82 cm and 4.1%, respectively. The average estimated time of the model was 52.14 ms. The accuracy of estimating crop plant height using this method was similar to that of existing methods. However, when estimating crop plant height used this method, once the estimation model was trained, the average and maximum crop plant height could be automatically estimated by inputting the pre-processed crop depth image, and the average estimation time could meet the real-time requirements of crop plant height acquisition. The relative errors in estimating average plant height and maximum plant height of oat showed a general decline trend with the increase of crop plant height. This might be because when crop plant height was low, the estimated results were more affected by soil fluctuation. The results of feature map visualization showed that the model could estimate plant height according to the height and contour of oat in depth image. Finally, an oat plant height estimation system was built based on 2023 Q1 version of LabVIEW software development platform. After depth camera acquiring oat depth images, the system could accurately estimate average and highest oat plant heights in real time without manual intervention, and the average estimation time was less than 0.1 seconds. The system could be used for crops irrigation and fertilization management. It could also be installed on the tractors to control the height of a sprinkler head during spraying, and to adjust the height of a cutting table during harvesting. 【Conclusion】 The depth image and double output regression convolutional neural network could be used to estimate oat plant height, and the accuracy could meet the production demand, so this method provided a basis for field management of oat crops.

Key words: depth image, deep learning, plant height, LabVIEW, oat