[发明专利]图像超分辨率重建方法及装置、存储介质

权利要求书

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

根据待重建原始图像进行数据生成处理，得到中间图像和第一彩色图像；

基于预设超分辨率神经网络对所述中间图像和所述第一彩色图像进行重构，得到重建后的图像；

其中，所述预设超分辨率神经网络包括预设图像恢复模型和预设颜色校正模型，所述基于预设超分辨率神经网络对所述中间图像和所述第一彩色图像进行重构，得到所述重建后的图像，包括：

将所述中间图像输入至所述预设图像恢复模型中，输出第一线性彩色图像；其中，所述第一线性彩色图像用于表征线性彩色图像预测；

将所述第一彩色图像输入至所述预设颜色校正模型中，输出全局校正参数和单像素校正参数；

根据所述第一线性彩色图像、所述全局校正参数以及所述单像素校正参数，获得所述重建后的图像。

2.根据权利要求1所述的方法，其特征在于，所述根据待重建原始图像进行数据生成处理，得到中间图像和第一彩色图像，包括：

对所述待重建原始图像进行预处理，获得所述中间图像；

对所述中间图像进行转换处理，获得所述第一彩色图像。

3.根据权利要求1所述的方法，其特征在于，所述根据待重建原始图像进行数据生成处理，得到中间图像和第一彩色图像之前，所述方法还包括：

获取所述待重建原始图像。

4.根据权利要求3所述的方法，其特征在于，所述获取所述待重建原始图像之后，所述方法还包括：

根据所述待重建原始图像进行数据生成处理，得到第二彩色图像和第二线性彩色图像；其中，所述第二彩色图像的分辨率高于所述第一彩色图像的分辨率。

5.根据权利要求4所述的方法，其特征在于，所述根据所述待重建原始图像进行数据生成处理，得到第二彩色图像，包括：

对所述待重建原始图像进行转换处理，获得所述第二彩色图像。

6.根据权利要求4所述的方法，其特征在于，所述根据所述中间图像和所述第一彩色图像，对所述待重建原始图像进行图像超分辨率重建，获得重建后的图像之后，所述方法还包括：

对所述第二彩色图像和所述重建后的图像进行对比，获得第一对比结果；

根据所述第一对比结果对所述预设超分辨率神经网络进行校正。

7.根据权利要求4所述的方法，其特征在于，所述将所述中间图像输入至所述预设图像恢复模型中，输出第一线性彩色图像之后，所述方法还包括：

对所述第一线性彩色图像和所述第二线性彩色图像进行对比，获得第二对比结果；

根据所述第二对比结果对所述预设超分辨率神经网络进行校正。

8.根据权利要求2所述的方法，其特征在于，所述对所述中间图像进行转换处理，获得所述第一彩色图像之后，所述方法还包括：

根据所述中间图像和所述预设图像恢复模型，获得第一图像特征；

根据所述第一彩色图像和所述预设颜色校正模型，获得第二图像特征；

根据所述第一图像特征和所述第二图像特征进行特征融合处理。

9.一种图像超分辨率重建装置，其特征在于，所述图像超分辨率重建装置包括：获取单元和重建单元，

所述获取单元，用于根据待重建原始图像进行数据生成处理，得到中间图像和第一彩色图像；

所述重建单元，用于基于预设超分辨率神经网络对所述中间图像和所述第一彩色图像进行重构，得到重建后的图像；

其中，所述预设超分辨率神经网络包括预设图像恢复模型和预设颜色校正模型；所述重建单元，还用于将将所述中间图像输入至所述预设图像恢复模型中，输出第一线性彩色图像；其中，所述第一线性彩色图像用于表征线性彩色图像预测；以及将所述第一彩色图像输入至所述预设颜色校正模型中，输出全局校正参数和单像素校正参数；以及根据所述第一线性彩色图像、所述全局校正参数以及所述单像素校正参数，获得所述重建后的图像。