有效

一种基于改进注意力机制的图像声呐小目标检测方法及系统

杨娟、何万钊、许枫、闫路
中国科学院声学研究所
杨娟机构 暂无
技术领域 暂无
何万钊机构 暂无
技术领域 暂无
许枫机构 暂无
技术领域 暂无
闫路机构 暂无
技术领域 暂无

摘要

本申请提供了一种基于改进注意力机制的图像声呐小目标检测方法及系统,基于YOLOv8实现,该方法包括:在YOLOv8的骨干网络的第10层插入如下过程:将原始图像进行上采样,再依次进行5x5点卷积、垂直卷积和水平卷积,最后经过激活函数得到数据F';将原始图像进行下采样,再依次进行1x1点卷积、水平卷积和垂直卷积,最后经过激活函数得到数据F”;将数据F'、数据F”和原始图像数据进行点乘,再与原始图像数据进行拼接,作为最后的输出结果。本申请的优势在于:方法通过上下采样让模型学习不同维度的图像特征,再通过水平和垂直方向的聚合,最后利用残差连接加速收敛,可以输出更精炼的结果,减少了背景干扰,减少了计算开销。