基于三目注意力机制的高效轻量化麦穗检测算法

作     者：董燕 刘运东 李卫杰 刘洲峰 李春雷 DONG Yan;LIU Yun-dong;LI Wei-jie;LIU Zhou-feng;LI Chun-lei

作者机构：中原工学院电子信息学院河南郑州450007 

基　　金：国家自然科学基金项目(U1804157、62072489、61772576) 河南高校科技创新团队基金项目(21IRTSTHN013) 

出 版 物：《计算机工程与设计》 (Computer Engineering and Design)

年 卷 期：2023年第44卷第7期

页      码：2062-2069页

摘      要：针对复杂大田环境下基于卷积神经网络的麦穗检测方法实时性差、检测精度低的问题,提出一种基于三目注意力机制的高效轻量化麦穗检测算法。采用MobileNetV3作为主干网络对特征进行高效提取;通过融合通道、空间和位置的特征信息设计三目注意力机制,提升模型对关键特征的敏感度;基于条件卷积设计全局多头自注意力机制,增强全局特征的提取能力;选择CIOU作为边框回归损失函数,增强遮挡及重叠目标的检测效果。实验结果表明,与其它优秀的算法相比,所提算法在减少模型容量的同时,提升了检测精度和速度。

主 题 词：复杂场景 麦穗检测 深度学习 轻量化 三目注意力机制 全局多头自注意力机制 条件卷积 

学科分类：081203[081203] 08[工学] 0835[0835] 0812[工学-测绘类] 

D　O　I：10.16208/j.issn1000-7024.2023.07.019

馆 藏 号：203122605...