摘要
为实现自然场景下苹果快速、精确检测与果实生长状态监测,提出基于改进的YOLOv4苹果检测模型与果径估测方法。针对自然场景下背景杂乱、目标较小等问题,在特征提取网络中引入注意力机制使模型更关注于果实区域,并在路径聚合网络中使用DO-Conv卷积以丰富网络提取的特征信息,提高模型检测性能。将检测后的目标区域进行CIELAB颜色空间分量下的阈值分割,利用RGB-D深度信息进行尺寸转换,实现果径估测。实验结果表明,模型检测精度达91.5%。果径估测平均绝对误差为1.91 mm,均方根误差(RMSE)为2.17 mm,总体分级准确率在90%以上,可为田间苹果分级与生长状态监测提供参考。
In order to realize fast and accurate apple detection and fruit growth status monitoring in natural scenes, an improved YOLOv4 apple detection model and fruit diameter estimation method are proposed. Aiming at the problems of cluttered background and small targets in natural scenes, the attention mechanism is introduced in the feature extraction network to make the model pay more attention to the fruit area, and DO-Conv convolution is used in the path aggregation network to enrich the feature information extracted by the network. Improve model checking performance. The detected target area is subjected to threshold segmentation under CIELAB color space components, and RGB-D depth information is used for size conversion to realize fruit diameter estimation. Experimental results show that the model detection accuracy reaches 91.5%. The average absolute error of fruit diameter estimation is 1.91 mm, the root mean square error(RMSE) is 2.17 mm, and the overall classification accuracy is above 90%, which can provide a reference for field apple grading and growth status monitoring.
作者
岳琳茜
李文宽
杨晓峰
李海芳
杨其晟
YUE Linxi;LI Wenkuan;YANG Xiaofeng;LI Haifang;YANG Qisheng(College of Information and Computer,Taiyuan University of Technology,Taiyuan 030600,China)
出处
《激光杂志》
CAS
北大核心
2022年第2期58-65,共8页
Laser Journal
基金
国家自然科学基金(No.61976150)
省重点研发计划项目(No.201803D31038)
山西省晋中市科技重点研发计划(No.Y192006)。
关键词
YOLOv4
苹果检测
注意力机制
果径测量
苹果分级
YOLOv4
apple detection
attention mechanism
fruit diameter measurement
apple grading
