有效

一种基于注意力融合的水下目标检测方法

张亚新、赵志强、唐金龙、吕帅帅、潘勉、于海滨
杭州电子科技大学

摘要

本发明公开了一种基于注意力融合的水下目标检测方法,包括以下步骤:步骤一,准备数据集。步骤二,构建基于注意力融合的目标检测网络。步骤三,获得通用场景下的特征提取网络。步骤四,构建损失函数,训练得到基于注意力融合的水下目标检测网络。本发明使用Vision‑Transformer模块提取输入图像的自注意力信息,并通过对输入图像分块,避免对完整图像计算自注意力,减少了计算量。使用PAFPN模块提取特征的空间注意力信息,并分层输出,将提取到的高级语义信息和低级细节信息进行融合。使用SE模块提取特征的通道注意力信息,进一步提升了模型的检测精度。