基于改进YOLO v7-tiny的小麦麦穗检测方法

打开文本图片集

doi：10.15889/j.issn.1002-1302.2024.20.018

摘要：针对农田环境下小麦麦穗目标检测精确率低的问题，在YOLO v7-tiny模型基础上进行深入改进，旨在提高麦穗检测的准确率，以满足农业生产管理系统和农业机器人边缘检测设备的需求。采用EfficientViT的主干网络替代YOLO v7-tiny的特征提取网络层，强化图像特征的提取能力；在特征融合网络层，引入CARAFE上采样模块替代原模型中的上采样模块，进一步优化特征融合过程；在特征融合网络层和输出层引入基于跨空间学习的高效多尺度注意力机制，有效提升模型的目标检测性能。（剩余12697字）