1.一种基于注意力融合的水下目标检测方法,其特征在于,步骤如下:步骤一,准备数据集;步骤二,构建基于注意力融合的目标检测网络;步骤三,获得通用场景下的特征提取网络;步骤四,构建损失函数,训练得到基于注意力融合的水下目标检测网络;步骤二具体方法如下:所述的基于注意力融合的目标检测网络包括主干特征提取网络、PAFPN路径增强特征金字塔模块、区域推荐网络和检测头;图片输入到目标检测网络中,通过主干特征提取网络提取自注意力信息,通过PAFPN路径增强特征金字塔模块提取空间注意力信息,通过检测头内部的SE模块提取通道间注意力信息;之后进行注意力信息融合,通过级联的方式传递特征,并根据不同类型的注意力机制的优势,对注意力信息进行融合,将主干特征提取网络提取的自注意力信息、PAFPN路径增强特征金字塔模块提取的空间注意力信息、检测头内部的SE模块提取的通道间注意力信息做逐步融合;目标检测网络使用Vision-Transformer作为主干特征提取网络,提取自注意力信息;所述的PAFPN路径增强的特征金字塔模块,包含特征金字塔模块和路径增强模块;特征金字塔模块通过下采样,压缩了特征的尺寸,提取了低级细节信息;路径增强模块通过上采样,扩大了特征的尺寸,提取了高级语义信息,并将低级细节信息和高级语义信息进行融合,并分M层输出;PAFPN路径增强的特征金字塔模块,不仅提取了多尺寸的特征信息,还将高级语义信息和低级细节信息进行融合,专注于空间层面的信息,提取了空间注意力信息;所述的区域推荐网络用于对PAFPN输出的每一层特征图做初步检测,检测出可能存在目标的区域并推荐给对应的检测头;区域推荐网络包括分类分支和定位分支两个分支;其中,分类分支是对区域内是否存在目标分类,若存在则将它的边界框推荐给检测头;定位分支是对目标所在区域做回归,输出目标所在边界框的左上角、右下角坐标;使用检测头内部的SE模块提取通道间注意力信息,并输送到检测头内部的定位分支和分类分支,对输入图像中可能存在的目标做检测;所述的检测头的个数由PAFPN路径增强的特征金字塔模块的层数确定,有M个;检测头根据送入的区域中可能存在目标的特征,对目标分类,并预测目标的位置;将特征拷贝,再输入到分类分支,经过全连接层输出目标属于可能类别的概率;将特征输入定位分支,经过全连接层输出目标可能所在边框的左上角、右下角的横纵坐标。
2.根据权利要求1所述的一种基于注意力融合的水下目标检测方法,其特征在于,步骤一具体方法如下:在实际水下场景下拍摄含有感兴趣目标的图片或视频,采用目标检测数据集标注软件为图片中出现的感兴趣目标做标注,获得水下目标检测数据集;从ImageNet官方网站下载数据集,以备模型预训练使用。
3.根据权利要求1所述的一种基于注意力融合的水下目标检测方法,其特征在于,步骤三具体方法如下:通过预训练数据集预训练基于注意力融合的目标检测网络的主干特征提取网络,得到具有强大特征提取能力的预训练模型权重。
4.根据权利要求3所述的一种基于注意力融合的水下目标检测方法,其特征在于,步骤四具体方法如下:构建位置回归损失函数和分类预测损失函数;其中位置回归损失函数采用smooth L1loss,衡量预测边界框与真实边界框之间的差距,分类预测损失函数采用Focal loss,衡量预测类别与真实类别之间的差距,其中,y取值为1或-1,表示目标是否是真实类别;p取值[0,1],表示目标是某待测类别的概率;α、γ用来调节分类损失的权重,α=0.25,γ=2;总损失函数是位置回归损失和分类预测损失之和:Loss=L reg +L class设计基于注意力融合的水下目标检测网络,采用Adam优化器更新模型权重,同时将多种注意力机制模型提取的特征进行融合,通过步骤一获得的水下目标检测数据集训练目标检测网络,得到基于注意力融合的水下目标检测网络;使用基于梯度下降的Adam优化算法更新水下目标检测网络模型权重;其中W t ,W t+1 分别表示在t阶段和t+1阶段的目标检测模型权重;η t 表示在t阶段的目标检测模型的学习率;m t ,m t-1 分别表示在t阶段和t-1阶段目标检测模型的一阶动量项;v t ,v t-1 分别表示在t阶段和t-1阶段目标检测模型的二阶动量项; 和 分别表示t阶段目标检测模型梯度的一阶矩和二阶矩;β 1 和β 2 分别表示一阶动量项和二阶动量项的常数系数,通常取0.9和0.999;∈是一个取值很小的数,为了避免分母为0。
5.根据权利要求1所述的一种基于注意力融合的水下目标检测方法,其特征在于,所述的PAFPN路径增强的特征金字塔模块中的M取值为5,PAFPN路径增强的特征金字塔模块分5层输出。