一种基于改进YOLOv7的声纳图像水下目标检测方法

一种基于改进YOLOv7的目标检测方法，包括以下步骤：获取水下声学图像；对水下声学图像进行数据增强，得到增强后的水下声学图像；将增强后的水下声学图像数据集划分为训练集和测试集；构建基于改进YOLOv7的声纳图像目标识别模型；基于训练集对基于改进YOLOv7的声纳图像目标识别模型进行训练，得到最优目标检测模型；将测试集输入所述最优目标检测模型中，实现对声纳图像的水下目标检测。本发明通过对YOLOv7模型进行改进，在主干提取网络引入注意力机制，提高模型对声纳图像复杂背景的抗干扰能力和目标特征的提取能力，同时在网络Neck部分，融入Multi GnBlocks模块实现关键特征之间的高阶交互。为提高算法的实时性，对模型冗余剪枝，极大压缩模型体积的同时保持检测的高精度。