[发明专利]深度图像的增强方法、装置、电子设备以及存储介质

说明书

本申请公开了一种深度图像的增强方法、装置、电子设备以及存储介质，涉及人工智能技术领域。该方法包括：获取彩色图像和待增强深度图像，彩色图像和待增强深度图像的图像内容相同，且彩色图像的分辨率大于待增强深度图像的分辨率，将待增强深度图像与彩色图像对齐，获得待增强深度图像对应的待增强平面图像，其中，彩色图像的分辨率与待增强平面图像的分辨率相同，将彩色图像和待增强平面图像输入已训练的神经网络模型，获得已训练的神经网络模型输出的已增强深度图像，已增强深度图像的分辨率大于待增强深度图像的分辨率。本申请通过彩色图引导深度图进行增强，提高深度图增强的准确性，节约了成本且丰富了深度图的应用场景，提升了实用性。

技术领域

本申请涉及人工智能技术领域，更具体地，涉及一种深度图像的增强方法、装置、电子设备以及存储介质。

背景技术

深度图像能够为立体场景的构建提供必要的深度信息，被广泛地运用于立体显示、机器视觉、虚拟现实等应用领域。但相比于彩色图像的获取，深度图像的获取需要使用成像密度更高的深度传感器，这种深度传感器通常价格非常昂贵，造成制作成本增加的问题。

发明内容

鉴于上述问题，本申请提出了一种深度图像的增强方法、装置、电子设备以及存储介质，以解决上述问题。

第一方面，本申请实施例提供了一种深度图像的增强方法，应用于电子设备，所述方法包括：获取彩色图像和待增强深度图像，其中，所述彩色图像和所述待增强深度图像的图像内容相同，且所述彩色图像的分辨率大于所述待增强深度图像的分辨率；将所述待增强深度图像与所述彩色图像对齐，获得所述待增强深度图像对应的待增强平面图像，其中，所述彩色图像的分辨率与所述待增强平面图像的分辨率相同；将所述彩色图像和所述待增强平面图像输入已训练的神经网络模型，获得所述已训练的神经网络模型输出的已增强深度图像，其中，所述已增强深度图像的分辨率大于所述待增强深度图像的分辨率。

第二方面，本申请实施例提供了一种深度图像的增强装置，应用于电子设备，所述装置包括：第一图像获得模块，用于获取彩色图像和待增强深度图像，其中，所述彩色图像和所述待增强深度图像的图像内容相同，且所述彩色图像的分辨率大于所述待增强深度图像的分辨率；第二图像获得模块，用于将所述待增强深度图像与所述彩色图像对齐，获得所述待增强深度图像对应的待增强平面图像，其中，所述彩色图像的分辨率与所述待增强平面图像的分辨率相同；图像增强模块，用于将所述彩色图像和所述待增强平面图像输入已训练的神经网络模型，获得所述已训练的神经网络模型输出的已增强深度图像，其中，所述已增强深度图像的分辨率大于所述待增强深度图像的分辨率。

第三方面，本申请实施例提供了一种电子设备，包括存储器和处理器，所述存储器耦接到所述处理器，所述存储器存储指令，当所述指令由所述处理器执行时所述处理器执行上述方法。

第四方面，本申请实施例提供了一种计算机可读取存储介质，所述计算机可读取存储介质中存储有程序代码，所述程序代码可被处理器调用执行上述方法。

本申请实施例提供的深度图像的增强方法、装置、电子设备以及存储介质，获取彩色图像和待增强深度图像，彩色图像和待增强深度图像的图像内容相同，且彩色图像的分辨率大于待增强深度图像的分辨率，将待增强深度图像与彩色图像对齐，获得待增强深度图像对应的待增强平面图像，其中，彩色图像的分辨率与待增强平面图像的分辨率相同，将彩色图像和待增强平面图像输入已训练的神经网络模型，获得已训练的神经网络模型输出的已增强深度图像，已增强深度图像的分辨率大于待增强深度图像的分辨率，从而通过彩色图引导深度图进行增强，提高深度图增强的准确性，节约了成本且丰富了深度图的应用场景，提升了实用性。

附图说明

为了更清楚地说明本申请实施例中的技术方案，下面将对实施例描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本申请的一些实施例，对于本领域技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其它的附图。