一种基于Transformer的水下图像增强方法

    公开(公告)号:CN114359091A

    公开(公告)日:2022-04-15

    申请号:CN202111645100.6

    申请日:2021-12-29

    Abstract: 本发明是一种基于Transformer的水下图像增强方法。本发明涉及数字图像处理和水下图像处理技术领域,本发明通过输入端输入待增强的水下图像,采用多级下采样模块通过卷积的方式进行下采样,获得不同尺度下的图像;通过全局特征提取模块采用卷积结合残差块的方式,进行不同尺度下图像的特征提取,获得不同尺寸下的特征图;将最小尺寸的特征图输入编码器,并采用多级特征深化提取的方式进行处理;将解码器的输出进行块解码,将最小尺寸特征图经过Transformer的输出从序列重新恢复成图像形式;经过多级上采样模块进行尺度上的恢复,结合之前的多级细节深化模块,获得输出的增强图像。

Patent Agency Ranking