[发明专利]超分辨率图像重建方法、装置、存储介质和电子设备

说明书

本发明实施例公开了一种超分辨率图像重建方法、装置、存储介质和电子设备，通过将低分辨率图像输入第一卷积层确定浅层特征图像序列，在将浅层特征图像输入多个顺序连接的特征提取模块，以确定多个对应的中间特征图像序列。根据各中间特征图像序列确定输入第一注意力模块的中间特征矩阵得到第一深层特征图像序列，以及输入第二注意力模块的第一中间图像序列得到第二深层特征图像序列。最后根据浅层特征图像序列和各深层特征图像序列确定高分辨率图像。本发明实施例可以通过超分辨率图像重建将低分辨率图像转换为高分辨率图像。同时，由于图像重建过程对中间层特征的重要性进行区分并加以利用，提高了输出的高分辨率图像的精度。

技术领域

本发明涉及计算机技术领域，尤其涉及一种超分辨率图像重建方法、装置、存储介质和电子设备。

背景技术

目前图像技术被广泛应用于各个领域，但当图像采集环境恶劣等情况下采集的图像分辨率过低时，无法直接使用。需要将低分辨率图像转换为高分辨率图像后使用。

发明内容

有鉴于此，本发明实施例提供了一种超分辨率图像重建方法、装置、存储介质和电子设备，旨在通过超分辨率图像重建方法将低分辨率图像转换为高分辨率图像。

第一方面，本发明实施例提供了一种超分辨率图像重建方法，所述方法包括：

将低分辨率图像输入第一卷积层，以确定浅层特征图像序列；

将所述浅层特征图像输入多个顺序连接的特征提取模块，以确定各所述特征提取模块对应的中间特征图像序列；

根据各所述中间特征图像序列确定中间特征矩阵和第一中间图像序列；

将所述中间特征矩阵输入第一注意力模块，以确定第一深层特征图像序列；

将所述第一中间图像序列输入第二注意力模块，以确定第二深层特征图像序列；

根据所述浅层特征图像序列、第一深层特征图像序列和第二深层特征图像序列确定高分辨率图像。

第二方面，本发明实施例提供了一种超分辨率图像重建装置，所述装置包括：

第一特征提取单元，用于将低分辨率图像输入第一卷积层，以确定浅层特征图像序列；

第二特征提取单元，用于将所述浅层特征图像输入多个顺序连接的特征提取模块，以确定各所述特征提取模块对应的中间特征图像序列；

中间图像确定单元，用于根据各所述中间特征图像序列确定中间特征矩阵和第一中间图像序列；

第一深层特征确定单元，用于将所述中间特征矩阵输入第一注意力模块，以确定第一深层特征图像序列；

第二深层特征确定单元，用于将所述第一中间图像序列输入第二注意力模块，以确定第二深层特征图像序列；

高分辨率图像确定单元，用于根据所述浅层特征图像序列、第一深层特征图像序列和第二深层特征图像序列确定高分辨率图像。

第三方面，本发明实施例提供了一种计算机可读存储介质，用于存储计算机程序指令，所述计算机程序指令在被处理器执行时实现如第一方面所述的方法。

第四方面，本发明实施例提供了一种电子设备，包括存储器和处理器，所述存储器用于存储一条或多条计算机程序指令，其中，所述一条或多条计算机程序指令被所述处理器执行以实现如第一方面所述的方法。