公开(公告)号：CN115984681A

公开(公告)日：2023-04-18

申请号：CN202211606325.5

申请日：2022-12-14

Applicant: 燕山大学

Inventor： 刘福才 ,  温个 ,  李绍宝

IPC: G06V20/05 ,  G06V10/40 ,  G06V10/82 ,  G06V10/80 ,  G06V10/774 ,  G06N3/0464 ,  G06N3/045 ,  G06N3/08

Abstract: 本发明公开了一种基于YOLOv5的水下目标检测方法，包括如下步骤：获取水下特定目标的初始数据集；对初始数据集进行增强，用以增加初始数据集的数量，得到扩充数据集；利用CA模块对主干网络里的部分C3模块进行改进；将SE注意力模块加入到主干网络的特定位置上；采用基于无锚盒的检测头对输入的图像进行检测；将扩充数据集输入到YOLOv5模型中，进行训练；用训练好的YOLOv5模型对水下图像进行检测。本发明能够保证在不大幅度降低检测速度的前提下，在复杂的水下环境中，提高检测模型对多种特定目标的检测精度。