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

说明书

本实施例公开一种图像超分辨率重建方法及装置、存储介质，该图像超分辨率重建方法可以包括：根据待重建原始图像进行数据生成处理，得到中间图像和第一彩色图像；根据中间图像和第一彩色图像，对待重建原始图像进行图像超分辨率重建，获得重建后的图像。

技术领域

本公开涉及计算机视觉领域，尤其涉及一种图像超分辨率重建方法及装置、存储介质。

背景技术

图像超分辨率问题是计算机视觉与图像增强领域的重要问题，然而，一方面由于真实的训练数据难以获取，且现有的数据生成方法简化了真实图片存在的噪声，模糊等问题，导致生成的数据同真实场景差距较大；另一方面，虽然相机能够同时提供原始图像和处理后的彩色图像等图像信息，但是现有技术并没有充分利用全部的图像信息，常常值使用彩色图像作为网络输入，却丢弃了原始图像中隐藏的信息。因此，现有的图像超分辨率技术在真实场景中表现平平，重建后的图像与真实图像的之间的差别较大。

发明内容

本实施例提供一种图像超分辨率重建方法及装置、存储介质，有效地提升重建后的图像逼近真实图像，图像质量高。

本公开的技术方案是这样实现的：

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

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

根据所述中间图像和所述第一彩色图像，对所述待重建原始图像进行图像超分辨率重建，获得重建后的图像。

在上述方案中，所述根据待重建原始图像进行数据生成处理，得到中间图像和第一彩色图像，包括：

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

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

在上述方案中，所述根据所述中间图像和所述第一彩色图像，对所述待重建原始图像进行图像超分辨率重建，获得重建后的图像，包括：

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

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

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

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

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

在上述方案中，所述根据待重建原始图像进行数据生成处理，得到中间图像和第一彩色图像之前，所述方法还包括：

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

在上述方案中，所述获取所述待重建原始图像之后，所述方法还包括：

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

在上述方案中，所述根据所述待重建原始图像进行数据生成处理，得到第二彩色图像，包括：

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

在上述方案中，所述根据所述中间图像和所述第一彩色图像，对所述待重建原始图像进行图像超分辨率重建，获得重建后的图像之后，所述方法还包括：

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