摘 要: 麦穗数量检测对于作物表型参数计算、产量预测和大田管理都具有重要的意义。为了解决人工计数工作量大且容易出错的问题,提出了一种基于You Only Look Once (YOLO)的麦穗目标检测与计数方法。首先利用大量小麦图像对深度神经网络进行训练,然后利用神经网络对小麦图像进行麦穗目标检测与计数,最后对神经网络目标检测的准确率和召回率进行计算评估,并通过分析检测结果验证其鲁棒性。分析结果显示,所训练网络对麦穗检测的精确率为76.96%,召回率为93.16%,均值平均精度mean Average Precision (mAP)为89.52%。此外,该模型可以检测不同生长时期的麦穗,具有较高的鲁棒性。研究表明,该方法对比其他麦穗计数方法准确高效,可以实际应用到小麦的产量估算上。 |
关键词: 目标检测;产量预测;YOLO;深度学习 |
中图分类号: TP391.4
文献标识码: A
|
基金项目: 南京林业大学青年科技创新基金“基于车载激光雷达的成熟小麦生物量在线估算方法研究”(CX2019018). |
|
Research of Wheat Ear Target Detection based on Convolutional Neural Network |
WANG Yuge, ZHANG Yong, HUANG Linxiong, ZHAO Fengkui
|
(College of Automobile and Traffic Engineering, Nanjing Forestry University, Nanjing 210037, China)
515400100@qq.com; zyjs111@126.com; 1773361196@qq.com; zfk@njfu.edu.cn
|
Abstract: Detecting the number of wheat ears is of great significance to the calculation of crop phenotypic parameters, yield prediction and field management. In order to solve the problem of heavy workload and error-prone manual counting, this paper proposes a wheat ear target detection and counting method based on You Only Look Once (YOLO). First, a large number of wheat images are used to train the deep neural network. Then, the neural network is used to detect and count wheat ears in the wheat images. Finally, the accuracy and recall rate of the neural network target detection are calculated and evaluated, and the robustness is verified by analyzing the detection results. The analysis results show that the trained network has an accuracy rate of 76.96% for wheat ear detection, a recall rate of 93.16%, and a mean Average Precision (mAP) of 89.52%. In addition, the model can detect wheat ears in different growth periods, and has high robustness. Studies have shown that this method is more accurate and efficient than other wheat ear counting methods, and can be applied to wheat yield estimation. |
Keywords: target detection; output prediction; YOLO; deep learning |