公开(公告)号：CN117876858A

公开(公告)日：2024-04-12

申请号：CN202410055010.9

申请日：2024-01-15

Applicant: 桂林电子科技大学

Inventor： 邓珍荣 ,  付亚威 ,  杨睿

IPC: G06V20/05 ,  G06V10/25 ,  G06V10/82 ,  G06V10/774 ,  G06N3/045 ,  G06N3/0455 ,  G06N3/0464 ,  G06N3/0499 ,  G06N3/082

Abstract: 本发明公开了一种基于改进yolov7算法的水下目标检测方法，该方法的步骤为：1）获取水下图像数据集，并按照一定的比例分为训练集、验证集和测试集；2）以yolov7模型为基础模型进行改进得到一种新的目标检测方法，改进方法包括对主干网络进行改进、设计一种新的模块、结合Transformer模型和使用新的损失函数；3）对改进后的模型进行剪枝；4）使用训练集对剪枝后的模型进行训练；5）使用测试集对模型进行评估。本发明方法具有较好的检测效果且减少了参数量和计算量，方法简单易部署，适用性广泛，尤其在水下机器人、海洋捕捞等领域将会被广泛应用。