[发明专利]图像深度信息的获取方法、装置、介质及电子设备

说明书

本申请的实施例提供了一种图像深度信息的获取方法、装置、介质及电子设备。该训练方法包括：根据待识别图像，获取所述待识别图像的图像特征；将所述图像特征输入至预训练的图像深度估计模型中，所述预训练的图像深度估计模型通过包含有光流信息、图像特征以及真实深度信息的样本数据训练得到；获取所述预训练的图像深度估计模型输出的所述待识别图像的深度信息。本申请实施例的技术方案通过包含有光流信息、图像特征以及真实深度信息的样本数据训练得到的图像深度估计模型可以降低深度估计结果的闪烁程度，保证深度估计结果的时域稳定性。

技术领域

本申请涉及计算机技术领域，具体而言，涉及一种图像深度信息的获取方法、装置、介质及电子设备。

背景技术

图像深度估计模型用于估计待处理的图像中各像素点的深度信息，获取待处理的图像的全局深度图，在计算机视觉和计算机图形学应用领域发挥着重要的作用。在目前的技术方案中，部分方法基于单个视频帧进行深度估计，然后该方法通常直接在单张图像上进行训练，其所估计的深度信息容易出现较大闪烁，且在时域中的稳定性较差。因此，如何降低深度估计结果的闪烁程度，保证深度估计结果的时域稳定性成为了亟待解决的技术问题。

发明内容

本申请的实施例提供了一种图像深度信息的获取方法、装置、介质及电子设备，进而至少在一定程度上可以降低深度估计结果的闪烁程度，保证深度估计结果的时域稳定性。

本申请的其他特性和优点将通过下面的详细描述变得显然，或部分地通过本申请的实践而习得。

根据本申请实施例的一个方面，提供了一种图像深度信息的获取方法，该方法包括：

根据待识别图像，获取所述待识别图像的图像特征；

将所述图像特征输入至预训练的图像深度估计模型中，所述预训练的图像深度估计模型通过包含有光流信息、图像特征以及真实深度信息的样本数据训练得到；

获取所述图像深度估计模型输出的所述待识别图像的深度信息。

根据本申请实施例的一个方面，提供了一种图像深度信息的获取装置，该装置包括：

第一获取模块，用于根据待识别图像，获取所述待识别图像的图像特征；

输入模块，用于将所述图像特征输入至预训练的图像深度估计模型中，所述预训练的图像深度估计模型通过包含有光流信息、图像特征以及真实深度信息的样本数据训练得到；

第二获取模块，用于获取所述图像深度估计模型输出的所述待识别图像的深度信息。

在本申请的一些实施例中，基于前述方案，该装置还包括训练模块，该训练模块包括：

输入单元，用于将内容相连续的第一训练样本和第二训练样本分别输入至学生模型中，得到所述学生模型输出的与所述第一训练样本对应的第一学生深度图像以及与所述第二训练样本对应的第二学生深度图像；

第一确定单元，用于根据所述第一训练样本和所述第二训练样本，确定所述第一训练样本与所述第二训练样本之间的光流信息；

第二确定单元，用于根据所述光流信息、所述第一学生深度图像以及所述第二学生深度图像，确定所述学生模型对应的时域损失信息；

第三确定单元，用于根据预先确定的所述第一训练样本的深度图像以及所述第一学生深度图像，确定所述学生模型对应的深度损失信息；

处理单元，用于根据所述时域损失信息以及所述深度损失信息，对所述学生模型进行调优处理，得到所述图像深度估计模型。