关键词:
番茄
果实计数
目标检测
多目标追踪
超深掩蔽
摘要:
针对在温室生产条件和作物种植模式约束下,番茄果实目标连续稳定跟踪困难,难以满足统计计数精度要求等问题,提出一种基于超深掩蔽与改进YOLOv8的不同成熟度果实计数方法。在YOLOv8基础上,构建融合全局特征的空间异质卷积核,优化设计卷积算子及目标检测网络,引入对果实目标标注更具鲁棒性的损失函数。提出用深度估计模型预测深度信息,动态生成深度阈值,基于该阈值,掩蔽远景果实目标,解决目标跟踪不稳定产生的计数精度低的问题。结果表明,与YOLOv8n相比,改进模型平均检测精度提高了3.2个百分点,召回率提高了3.7个百分点;将所设计的卷积算子用于目标检测模型,与使用该算子前相比,果实检测精度提高了2.7个百分点,与引入鲁棒性损失函数前相比,引入该损失后,检测精度提高了1.1个百分点;与不用超深掩蔽处理相比,应用该处理后,番茄果实计数精度提高了12.63个百分点;该方法的番茄果实计数精度为93.80%,对不同成熟度果实的计数精度不低于91.00%,计算速度为23帧/s。对YOLOv8的改进是有效的,超深掩蔽对提高番茄计数精度具有重要作用,研究可为基于视觉技术的果蔬产量统计提供技术参考。