|国家科技期刊平台
首页|期刊导航|农业机械学报|基于轻量化U-Net网络的果园垄间路径识别方法

基于轻量化U-Net网络的果园垄间路径识别方法OA北大核心CSTPCD

Path Recognition Method of Orchard Ridges Based on Lightweight U-Net

中文摘要英文摘要

针对目前果园垄间导航路径识别方法存在准确性与实时性难以同时兼顾、泛化能力弱等问题,本文在U-Net模型的基础上进行优化,采用MobileNet-v3 Large作为U-Net的主干特征提取网络,并在跳跃连接处引入坐标注意力机制(Coordinate attention,CA),构建轻量化路径识别模型.以该模型分割的垄间可行驶区域为基础,利用最小二乘法重塑可行驶区域边缘点,并进一步提取垄间导航线.首先采用数据增强的草莓垄间数据集进行模型训练,并进一步迁移到葡萄和蓝莓数据集上进行权重微调,以提高模型适应能力.最后在相应的验证集上进行导航路径识别,并通过可视化对比不同模型识别结果,以验证模型准确性.试验结果表明,网络模型在草莓、蓝莓和葡萄果园垄间路径识别的平均交并比分别为98.06%、97.36%和98.50%,平均像素准确度分别达到99.13%、98.75%和99.29%.模型处理RGB图像分割可行驶区域的理论推理速度可达19.23 f/s,满足导航实时性和准确性的要求.

In response to the issues of accuracy and speed being difficult to balance simultaneously,as well as the weak generalization ability in the current navigation path recognition methods of fruit ridges,an optimization approach was proposed based on the U-Net model.The optimization involved integrating MobileNet-v3 Large as the backbone feature extraction network for U-Net and introducing coordinate attention at the skip connections to construct a lightweight path recognition model.Based on the drivable area segmented by this model in the inter-ridge,the edge points of the area were reshaped by using the least squares method,and further the inter-ridge navigation lines were extracted.Firstly,the model was trained on the augmented strawberry interrow dataset,and further migrated to the grape and blueberry datasets for weight fine-tuning to improve the model's adaptability.Finally,the navigation path was identified on the corresponding verification set,and the recognition results of different models were compared visually to verify the accuracy of the model.Experimental results demonstrated that the model achieved an average intersection over union of 98.06%,97.36%,and 98.50%for strawberry,blueberry,and grape interridge navigation path segmentation accuracy respectively,and the average pixel accuracy reached 99.13%,98.75%,and 99.29%.The theoretical reasoning speed of the model for segmenting of RGB images was up to 19.23 f/s,and the average time from image input to completed path extraction was 0.211 s,meeting the requirements of real-time navigation and accuracy.A method of path extraction based on semantic segmentation was proposed,which provided a general method for the navigation of agricultural machinery equipment in interridge operation.

侯文慧;周传起;程炎;王玉伟;刘路;秦宽

安徽农业大学工学院,合肥 230036

计算机与自动化

垄间导航路径识别语义分割图像处理迁移学习U-Net

ridge navigationpath recognitionsemantic segmentationimage processingtransfer learningU-Net

《农业机械学报》 2024 (002)

16-27 / 12

安徽省高校科研计划项目(2022AH050872)和电气传动与控制安徽省重点实验室开放基金项目(DQKJ202203)

10.6041/j.issn.1000-1298.2024.02.002

评论