基于注意力机制和多空间金字塔池化的实时目标检测算法

基于注意力机制和多空间金字塔池化的实时目标检测算法王国刚;李泽欣;董志豪
【期刊名称】《计算机测量与控制》
【年(卷),期】2024(32)2
【摘要】YOLOv4计算复杂度高,空间金字塔池化模块仅一次增强特征融合网络的深层区域特征图的表征能力、检测头网络的特征图难以突出重要通道特征;针对以上问题,提出一种基于注意力机制和多空间金字塔池化的实时目标检测算法;该算法采用多空间金字塔池化,提取局部特征和全局特征,融合多重感受野,加强特征融合网络的浅、中、深层特征图的表征能力;引入压缩激励通道注意力机制,建模通道间的相关性,自适应调整特征图各个通道的权重,从而使网络更加关注重要特征;特征融合和检测头网络中使用深度可分离卷积,减少了网络参数量;实验结果表明,所提算法的均值平均精度均高于其他7种主流对比算法;与YOLOv4相比,参数量、模型大小分别减少了27.85 M和106.25 MB,所提算法在降低复杂度的同时,提高了检测准确度,且该算法的检测速率达到33.70帧/秒,满足实时性要求。

【总页数】9页(P56-64)
【作者】王国刚;李泽欣;董志豪
【作者单位】山西大学物理电子工程学院
【正文语种】中文
【中图分类】TP391
【相关文献】
1.基于注意力机制与空间金字塔池化的行人属性识别
2.基于条形池化和注意力机制的街道场景红外目标检测算法
3.基于通道注意力机制与金字塔池化的包裹破损检测算法
4.基于空洞空间卷积池化金字塔结构和注意力机制的全卷积残差网络磨玻璃肺结节分割方法
5.基于密集空洞空间金字塔池化和注意力机制的皮肤病灶图像分割方法
因版权原因，仅展示原文概要，查看原文内容请购买。