[发明专利]图像重建方法、装置、电子设备及计算机可读存储介质

说明书

本申请实施例提供了一种图像重建方法、装置、电子设备及计算机可读存储介质，涉及人工智能、计算机视觉技术、智慧交通和云技术领域。该方法包括：获取目标对象对应的待处理图像，获取待处理图像的第一纹理信息和第一几何信息。对待处理图像进行风格特征提取，得到待处理图像的第一风格特征。采用第一风格特征对第一纹理信息进行修复，得到第二纹理信息；基于第二纹理信息和第一几何信息进行图像重建，得到重建后的目标图像。该方法基于对待处理图像进行风格特征提取所得到第一风格特征，对第一纹理信息进行修复，得到第二纹理信息。并根据该第二纹理信息和第一几何信息进行图像重建，可以得到纹理信息更加丰富的目标图像。

技术领域

本申请涉及人工智能、计算机视觉技术、智慧交通和云技术领域，具体而言，本申请涉及一种图像重建方法、装置、电子设备、计算机可读存储介质及计算机程序产品。

背景技术

相关技术中，在进行图像重建时，通常采用参数化方法或非参数化方法。在采用参数化方法执行图像重建任务时，耗时较长，重建结果通常过于平滑，缺少细节以及其他显著的结构特征，且难以进行大规模的图像重建任务。在采用非参数化方法执行图像重建任务时，由于没有预先定义的拓扑结构，所得到的图像容易受到噪声和姿态的干扰，且无法实现将低质量图像重建为高质量图像。

基于上述，如何大规模地实现将低质量图像重建为高质量图像，有效提高待处理图像的图像质量，成为了本领域亟待解决的技术问题。

发明内容

本申请实施例提供了一种图像重建方法、装置、电子设备、计算机可读存储介质及计算机程序产品，该方法可以实现将目标对象对应的低质量图像重建为对应的高质量图像，有效提高了待处理图像的图像质量。

根据本申请实施例的一个方面，提供了一种图像重建方法，该方法包括：

获取目标对象对应的待处理图像；

获取待处理图像的初始图像信息，该初始图像信息包括第一纹理信息和第一几何信息；

基于待处理图像和第一纹理信息，通过训练好的图像重建模型执行以下操作，得到第二纹理信息：

对待处理图像进行风格特征提取，得到待处理图像的第一风格特征；采用第一风格特征对第一纹理信息进行修复，得到第二纹理信息；

基于第二纹理信息和第一几何信息进行图像重建，得到待处理图像对应的目标图像，其中，目标图像的图像质量高于待处理图像的图像质量。

根据本申请实施例的另一个方面，提供了一种图像重建装置，该装置包括图像获取模块、图像信息获取模块、纹理修复模块和图像重建模块，其中：

图像获取模块，用于获取目标对象对应的待处理图像；

图像信息获取模块，用于获取待处理图像的初始图像信息，初始图像信息包括第一纹理信息和第一几何信息；

纹理修复模块，用于基于待处理图像和第一纹理信息，通过训练好的图像重建模型执行以下操作，得到第二纹理信息：

对待处理图像进行风格特征提取，得到待处理图像的第一风格特征；采用第一风格特征对第一纹理信息进行修复，得到第二纹理信息；

图像重建模块，用于基于第二纹理信息和第一几何信息进行图像重建，得到待处理图像对应的目标图像，其中，目标图像的图像质量高于待处理图像的图像质量。

可选地，图像重建模块在基于第二纹理信息和第一几何信息进行图像重建，得到待处理图像对应的目标图像时，具体用于：

采用第二纹理信息，通过图像重建模型对第一几何信息进行修复，得到第二几何信息；

基于第二纹理信息和第二几何信息进行图像重建，得到目标图像。