[发明专利]一种图像重建方法及装置

说明书

本申请实施例提供了一种图像重建方法及装置，包括：获得待处理图像；对待处理图像进行特征提取得到第一特征图像；以蒸馏输入图像为蒸馏对象进行蒸馏处理得到蒸馏输出图像；若第一数量未达到第一预设数量重复以蒸馏输入图像和已得蒸馏输出图像为蒸馏对象进行蒸馏处理，得到新的蒸馏输出图像，直至第一数量达到第一预设数量；若第二数量未达到第二预设数量将蒸馏输入图像更新为最新得到的蒸馏输出图像，返回以蒸馏输入图像为蒸馏对象进行蒸馏处理的步骤，直至第二数量达到第二预设数量；根据最新得到的蒸馏输出图像进行图像重建，得到分辨率为重建分辨率的重建图像。应用本申请实施例提供的方案进行图像重建，能够提高所得高分辨率图像的图像质量。

技术领域

本申请涉及图像处理技术领域，特别是涉及一种图像重建方法及装置。

背景技术

由于HR(High Resolution，高分辨率)图像相对于LR(Low Resolution，低分辨率)图像具有像素密度高、细节信息丰富、画质细腻、辨识度高等优点，数字高清、卫星成像、视频监控、虚拟现实和生物医学图像分析等应用往往基于HR图像实现。

上述HR图像可以通过高分辨率相机或者超分辨率相机进行图像采集得到。但是考虑到上述相机成本较高、HR图像对传输带宽以及存储资源的需求较高等因素，大多数应用通过SR(Super Resolution，超分辨率)技术对LR图像进行重建得到HR图像。其中，SR技术是指通过算法将图像的分辨率由LR提升到HR的技术。

现有技术中，常用的SR技术是基于插值算法实现的。也就是，对LR图像进行插值处理得到HR图像。虽然应用上述方式能够对LR图像进行重建得到HR图像，但是插值处理会导致图像的细节信息丢失，从而导致得到的HR图像质量差。参见图1a和图1b，其中，图1a示出了一张原始图像，图1b示出了对图1a进行二分之下采样得到LR图像后，再对LR图像进行插值处理得到的与上述原始图像分辨率相同的HR图像。显然图1b相对于图1a损失了大量细节信息。

发明内容

本申请实施例的目的在于提供一种图像重建方法及装置，以提高对图像进行重建得到的HR图像的图像质量。具体技术方案如下：

第一方面，本申请实施例提供了一种图像重建方法，所述方法包括：

获得待处理图像；

对所述待处理图像进行特征提取，得到第一特征图像；

以蒸馏输入图像为蒸馏对象进行蒸馏处理，得到蒸馏输出图像，其中，所述蒸馏输入图像的初始图像为所述第一特征图像；

在第一数量未达到第一预设数量的情况下，重复以所述蒸馏输入图像和已得蒸馏输出图像为蒸馏对象进行蒸馏处理，得到新的蒸馏输出图像，直至所述第一数量达到第一预设数量，其中，所述已得蒸馏输出图像包括：最近一次执行第一蒸馏处理后、且在本次执行第二蒸馏处理前得到的所有蒸馏输出图像，所述第一数量为：最近一次执行所述第一蒸馏处理后执行所述第二蒸馏处理的次数，所述第一蒸馏处理为：蒸馏对象为所述蒸馏输入图像的蒸馏处理，所述第二蒸馏处理为：蒸馏对象为所述蒸馏输入图像和已得蒸馏输出图像的蒸馏处理；

在第二数量未达到第二预设数量的情况下，将所述蒸馏输入图像更新为最新得到的蒸馏输出图像，并返回所述以蒸馏输入图像为蒸馏对象进行蒸馏处理的步骤，直至所述第二数量达到第二预设数量，其中，所述第二数量表示：执行所述第一蒸馏处理的次数；

根据最新得到的蒸馏输出图像进行图像重建，得到分辨率为重建分辨率的重建图像，其中，所述重建分辨率为：高于所述待处理图像的分辨率的预设分辨率。

本申请的一个实施例中，通过以下方式进行蒸馏处理：

对作为蒸馏对象的特征图像进行特征维度转换，得到第三预设数量张第二特征图像；

通过卷积操作对所述第二特征图像进行特征增强处理，得到第三特征图像；