2023年7月第44卷第7期计算机工程与设计COMPUTERENGINEERINGANDDESIGNJuly2023Vol.44■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■No.7基于三目注意力机制的高效轻量化麦穗检测算法董燕,刘运东,李卫杰,刘洲峰,李春雷+(中原工学院电子信息学院,河南郑州450007)摘要:针对复杂大田环境下基于卷积神经网络的麦穗检测方法实时性差、检测精度低的问题,提出一种基于三目注意力机制的高效轻量化麦穗检测算法。采用MobileNetV3作为主干网络对特征进行高效提取;通过融合通道、空间和位置的特征信息设计三目注意力机制,提升模型对关键特征的敏感度;基于条件卷积设计全局多头自注意力机制,增强全局特征的提取能力;选择CIOU作为边框回归损失函数,增强遮挡及重叠目标的检测效果。实验结果表明,与其它优秀的算法相比,所提算法在减少模型容量的同时,提升了检测精度和速度。关键词:复杂场景;麦穗检测;深度学习;轻量化;三目注意力机制;全局多头自注意力机制;条件卷积中图法分类号:TP391文献标识号:A文章编号:1000-7024(2023)07-2062-08doi:10.16208/j.issn1000-7024.2023.07.019收稿日期:2021-12-28;修订日期:2023-07-03基金项目:国家自然科学基金项目(U1804157、62072489、61772576);河南高校科技创新团队基金项目(21IRTSTHN013)作者简介:董燕(1977),女,河南兰考人,硕士,副教授,研究方向为计算机视觉;刘运东(1997),男,河南驻马店人,硕士研究生,研究方向为图像处理与目标检测;李卫杰(1996),女,河南周口人,硕士研究生,研究方向为信息处理;刘洲峰(1962),男,河南新乡人,博士,教授,研究方向为图像处理与模式识别;+通讯作者:李春雷(1979),男,河南周口人,博士,教授,CCF会员,研究方向为计算机视觉。E-mail:lichunlei1979@sina.comEfficientandlightweightwheateardetectionalgorithmwithtrinocularattentionmechanismDONGYan,LIUYun-dong,LIWe...