公开(公告)号：CN116596788A

公开(公告)日：2023-08-15

申请号：CN202310551602.5

申请日：2023-05-16

申请人： 大连海事大学

发明人： 付先平 ,  张竣博 ,  曹楠 ,  覃文强 ,  王亚飞

IPC分类号： G06T5/00 ,  G06T5/50 ,  G06V10/80 ,  G06V10/82 ,  G06V20/05 ,  G06N3/0455 ,  G06N3/0464 ,  G06N3/0985

摘要： 本发明公开了一种基于Transformer的多阶段水下图像增强方法，该方法包含如下步骤：建立图像训练集，获取参考图像和待处理的水下图像；构建基于Transformer骨干的多阶段水下图像增强网络；构建损失函数，对基于Transformer骨干的多阶段水下图像增强网络的训练加以约束，得到训练好的多阶段水下图像增强网络；将参考图像和待处理的水下图像依次输入到训练好的多阶段水下图像增强网络中，实现待处理的水下图像增强处理，本发明能够针对不同水下场景下，有效地改善水下因后向散射导致的色偏和对比度低的问题，同时增强了图像的细节和纹理，提高水下降质图像的视觉感知效果。