基于多尺度混合注意力模块的不良天气下目标检测方法

    公开(公告)号:CN118334413A

    公开(公告)日:2024-07-12

    申请号:CN202410387028.9

    申请日:2024-04-01

    Applicant: 东南大学

    Abstract: 本发明公开了一种基于多尺度混合注意力模块的不良天气下目标检测方法,首先使用主干网络对输入图像进行特征提取,接着将主干网络输出的特征图送入多尺度混合注意力模块,最后送入检测头获得检测结果。其中多尺度混合注意力模块包含三个部分,轻量级多尺度特征融合模块、通道注意力模块和空间注意力模块。本发明的多尺度混合注意力模块可以捕获图像不同尺度的特征,在通道和空间维度上同时实现了注意力聚焦,并且参数量不大,十分轻量,可以增强关键特征的表示,同时抑制背景噪声和干扰,帮助模型捕捉和理解在不良天气条件下的场景细节,对提高不良天气条件下的目标检测精度有着重要意义。

Patent Agency Ranking