[发明专利]视差数据生成方法、装置、计算机设备及存储介质

说明书

本申请公开了一种视差数据生成方法、装置、计算机设备及存储介质，属于计算机视觉技术领域。该方法包括：获取至少一组原始图像对，原始图像对中包含原始左目图像和原始右目图像；对原始图像对进行光流提取，得到各组原始图像对对应的光流值；基于光流值对原始图像对进行光流筛选，得到目标单目图像和目标单目图像中各个像素点对应的目标光流值，目标单目图像为原始左目图像或原始右目图像；基于目标光流值，生成目标单目图像对应的目标视差数据。提高了视差数据的获取效率，基于该视差数据可以快速构建大量用于训练单目深度估计模型的预训练样本，进而可以提高深度估计模型的训练效率。

技术领域

本申请实施例涉及计算机视觉技术领域，特别涉及一种视差数据生成方法、装置、计算机设备及存储介质。

背景技术

基于图像的深度信息估计是计算机视觉领域的基础性任务之一，在3维重建、增强现实(Augmented Reality，AR)交互等领域有广泛应用。基于深度信息估计所使用图像数量的不同，深度估计的相关算法可以分为多目图像深度估计和单目图像深度估计两大类。其中，单目深度估计算法有着更广阔的应用前景，在单目图像深度估计领域中，主要通过训练网络模型的方式实现深度信息估计。

相关技术中，在训练用于进行深度信息估计的网络模型时，往往需要获取到足够多的训练样本，所需要的训练样本形式为：单目图像和配套的单通道深度谱，深度谱上每个位置的取值代表原图对应位置与相机的距离。一般的获取方式为通过专门的传感器采集图像，以获取到具有深度信息的图像。但是该获取方式获取成本较高，且获取效率较低，无法用于快速获取到具有深度信息的图像。

发明内容

本申请实施例提供了一种视差数据生成方法、装置、计算机设备及存储介质。所述技术方案如下：

一方面，本申请实施例提供了视差数据生成方法，所述方法包括：

获取至少一组原始图像对，所述原始图像对中包含原始左目图像和原始右目图像，所述原始左目图像和所述原始右目图像是在左目视角和右目视角下观察同一场景所对应的图像；

对所述原始图像对进行光流提取，得到各组所述原始图像对对应的光流值；

基于所述光流值对所述原始图像对进行光流筛选，得到目标单目图像和所述目标单目图像中各个像素点对应的目标光流值，其中，光流筛选后各个像素点对应目标光流值之间的相对大小关系与视差数据之间的相对大小关系相同，所述目标单目图像为原始左目图像或原始右目图像；

基于所述目标光流值，生成所述目标单目图像对应的目标视差数据。

另一方面，本申请实施例提供了一种一种视差数据生成装置，所述装置包括：

获取模块，用于获取至少一组原始图像对，所述原始图像对中包含原始左目图像和原始右目图像，所述原始左目图像和所述原始右目图像是在左目视角和右目视角下观察同一场景所对应的图像；

光流提取模块，用于对所述原始图像对进行光流提取，得到各组所述原始图像对对应的光流值；

光流筛选模块，用于基于所述光流值对所述原始图像对进行光流筛选，得到目标单目图像和所述目标单目图像中各个像素点对应的目标光流值，其中，光流筛选后各个像素点对应目标光流值之间的相对大小关系与视差数据之间的相对大小关系相同，所述目标单目图像为原始左目图像或原始右目图像；

生成模块，用于基于所述目标光流值，生成所述目标单目图像对应的目标视差数据。

另一方面，本申请实施例提供了一种计算机设备，所述计算机设备包括处理器和存储器，所述存储器中存储有至少一条程序代码，所述程序代码由所述处理器加载并执行以实现如上述方面所述的视差数据生成方法。