公开(公告)号：CN117745610A

公开(公告)日：2024-03-22

申请号：CN202311683096.1

申请日：2023-12-08

Applicant: 大连海事大学

Inventor： 张维石 ,  张得欢 ,  周景春 ,  武照林

IPC: G06T5/90 ,  G06N3/0464 ,  G06N3/08 ,  G06V10/44 ,  G06V10/80 ,  G06V10/82 ,  G06N3/048

Abstract: 本发明提供了一种基于多方向卷积注意机制的水下图像增强方法，包括以下步骤：对于输入图像使用特征提取模块进行初步特征，然后输入到编码器中，其中包括使用下采样操作、多方向卷积注意机制提取大感受野的特征，多方向卷积注意机制结合四个方向的特征进行增强。重复上述操作四次，获取不同尺寸的特征信息。针对最小尺寸特征使用多方向卷积注意机制进行特征增强，输入到解码器中进行特征恢复。其中编码器包括使用上采样操作恢复特征尺寸，结合编码器中对应尺度的特征进行特征恢复，重复上述操作四次。最后使用平滑L1损失函数、感知损失函数和多尺度结构相似性损失函数对于网络进行引导，有效增强水下图像。