[发明专利]图像超分辨率重建方法及装置

权利要求书

1.一种图像超分辨率重建方法，其特征在于，包括：

确定待重建图像；

将所述待重建图像输入至图像重建模型，得到所述图像重建模型输出的超分辨率图像；

其中，所述图像重建模型是基于样本图像以及样本图像对应的低分辨率图像训练得到的；所述图像重建模型通过注意力层对所述待重建图像的各子图像的初始特征进行注意力特征提取，并基于各子图像的注意力特征进行超分辨率重建，所述注意力层包括多个注意力模块，各注意力模块用于对上一注意力特征依次进行上采样和注意力特征提取，得到当前注意力特征，并将所述当前注意力特征作为下一注意力模块的上一注意力特征；首个注意力模块的上一注意力特征为所述待重建图像的各子图像的初始特征，末个注意力模块的当前注意力特征作为各子图像的注意力特征；

所述将所述待重建图像输入至图像重建模型，得到所述图像重建模型输出的所述待重建图像的超分辨率图像，包括：

将所述待重建图像输入至所述图像重建模型的图像分割层，由所述图像分割层对所述待重建图像进行切分，得到所述图像分割层输出的所述待重建图像的多个子图像；

将各子图像输入至所述图像重建模型的特征嵌入层，由所述特征嵌入层将各子图像的RGB值按顺序展开为多个嵌入向量，得到所述特征嵌入层输出的多个嵌入向量；

将多个嵌入向量输入至所述图像重建模型的特征提取层，由所述特征提取层对多个嵌入向量进行注意力特征提取，得到所述特征提取层输出的各子图像的初始特征；

将各子图像的初始特征输入至所述注意力层，得到所述注意力层输出的各子图像的注意力特征；

将各子图像的注意力特征输入至所述图像重建模型的图像重建层，由所述图像重建层对各子图像的注意力特征进行卷积处理，得到所述图像重建层输出的所述待重建图像的超分辨率图像。

2.根据权利要求1所述的图像超分辨率重建方法，其特征在于，所述将各子图像的初始特征输入至所述注意力层，得到所述注意力层输出的各子图像的注意力特征，包括：

将所述上一注意力特征输入至当前注意力模块，得到所述当前注意力模块输出的所述当前注意力特征，并将所述当前注意力特征作为所述下一注意力模块的上一注意力特征。

3.根据权利要求1所述的图像超分辨率重建方法，其特征在于，当前注意力模块包括上采样子层和注意力子模块；所述注意力子模块包括一个第一注意力子层，或包括至少一个第一注意力子层和至少一个第二注意力子层；所述第一注意力子层与所述第二注意力子层交替排列；

所述上采样子层用于对所述上一注意力特征进行上采样，得到上采样特征，并将所述上采样特征作为首个第一注意力子层的输入；

所述第一注意力子层用于对上一子层输出的特征进行多头注意力特征提取，所述第二注意力子层用于基于移位窗口对上一子层输出的特征进行多头注意力特征计算。

4.根据权利要求1所述的图像超分辨率重建方法，其特征在于，所述特征提取层至少包括一个第一子层和一个第二子层，所述第一子层与所述第二子层交替排列；

所述第一子层用于对多个嵌入向量进行注意力特征提取，得到初始子图像注意力特征；所述第二子层用于基于移位窗口对所述初始子图像注意力特征进行多头注意力特征计算，得到各子图像的初始特征。

5.根据权利要求1至4任一项所述的图像超分辨率重建方法，其特征在于，各注意力模块是基于目标算法对上一注意力特征进行上采样的，所述目标算法为双线性插值法、最近邻插值法以及双立方插值法中的任一种。