关键词:
深度学习
目标检测
人参果成熟度
注意力机制
轻量化网络
摘要:
随着农业智能化的发展,果实采摘机器人在水果采收领域的应用日益广泛,果实成熟度检测是决定采摘效率与质量的关键环节。然而,目前实现人参果成熟度自动化检测,面临缺乏公开数据集、自然环境干扰检测精度以及检测模型难以轻量化部署等难题。针对这些问题,本文采用深度学习技术,从数据集构建、检测精度优化、轻量化模型设计等方面对人参果成熟度检测进行了深入研究,为人参果智能化采摘提供有力的技术支撑。主要研究工作如下:
(1)人参果成熟度数据集的构建。本研究采用相机与智能手机采集不同成熟度及遮挡条件下的果园图像,并制定成熟度等级划分标注,将其划分为成熟期、过渡期和未成熟期三类。利用Label Img工具进行边界框标注,确保标注与成熟度等级一致。采用自动色彩均衡算法(Automatic Color Equalization,ACE)增强图像色彩与对比度,并通过翻转、错切、亮度调整与增加椒盐噪声进行数据扩增。最终按7:2:1比例划分为训练集、验证集与测试集,为模型训练与评估提供数据支持。
(2)检测精度提升设计。为提升果实遮挡及小目标检测场景下的人参果成熟度检测精度,提出一种改进YOLOv7的人参果成熟度检测模型。首先,引入混合注意力机制(Mixture of self-attention and convolution,ACmix),使模型能够自适应地聚焦于遮挡区域以及人参果表面关键纹理特征,显著增强特征提取的针对性。其次,采用内容感知特征重组上采样(Content-Aware Re Assembly of Feature Maps,CARAFE)上采样算子,提升低分辨率小目标及遮挡目标的边界识别精度,避免信息丢失。最后,优化边界框回归策略,以Focaler-GIo U替换原始损失函数,提高遮挡目标的定位精度。实验结果表明:改进后模型的精确率、召回率和平均精度均值分别达到92.4%、90.0%和95.3%,比原始模型提升1.9%、0.7%和2%,有效增强了复杂环境下的检测能力。
(3)轻量化模型设计。鉴于人参果成熟度检测模型普遍存在参数量大、计算复杂度高的问题,难以在资源受限的农业设备上高效运行,提出一种基于改进YOLOv8n的轻量化人参果成熟度检测模型。首先,在YOLOv8n主干网络中引入Ghost Net网络中的C3Ghost和Ghost Conv模块,并融合轻量级十字交叉注意力模块,在降低模型参数量计算量成本的同时,增强关键特征提取能力。其次,引入渐进空间金字塔结构(Asymptotic Feature Pyramid Network,AFPN),提升不同尺度人参果的检测精度。实验结果表明:改进后模型的计算量减少33.3%,参数量减少50.8%,平均精度均值达到94.3%,实现了检测性能与资源消耗的平衡,为人参果智能化检测与采摘的实际应用提供了技术支撑。
(4)人参果成熟度检测系统实现。根据人参果成熟度检测任务需求,基于Py Qt5框架开发了智能检测系统。该系统实现了单张/批量图像的可视化检测功能,能够快速识别并标注人参果成熟度等级,为果园精准化管理提供了可靠的技术支持。