[發明專利]視差數據生成方法、裝置、計算機設備及存儲介質

在審

• 申請號: 202110258449.8
• 申請日: 2021-03-09
• 公開（公告）號: CN112991419A
• 公開（公告）日: 2021-06-18
• 發明（設計）人: 尹康
• 申請（專利權）人: OPPO廣東移動通信有限公司
• 主分類號: G06T7/55
• 分類號: G06T7/55
• 代理公司: 北京三高永信知識產權代理有限責任公司 11138
• 代理人: 邢惠童

本申請公開了一種視差數據生成方法、裝置、計算機設備及存儲介質，屬于計算機視覺技術領域。該方法包括：獲取至少一組原始圖像對，原始圖像對中包含原始左目圖像和原始右目圖像；對原始圖像對進行光流提取，得到各組原始圖像對對應的光流值；基于光流值對原始圖像對進行光流篩選，得到目標單目圖像和目標單目圖像中各個像素點對應的目標光流值，目標單目圖像為原始左目圖像或原始右目圖像；基于目標光流值，生成目標單目圖像對應的目標視差數據。提高了視差數據的獲取效率，基于該視差數據可以快速構建大量用于訓練單目深度估計模型的預訓練樣本，進而可以提高深度估計模型的訓練效率。

技術領域

本申請實施例涉及計算機視覺技術領域，特別涉及一種視差數據生成方法、裝置、計算機設備及存儲介質。

背景技術

基于圖像的深度信息估計是計算機視覺領域的基礎性任務之一，在3維重建、增強現實(Augmented Reality，AR)交互等領域有廣泛應用。基于深度信息估計所使用圖像數量的不同，深度估計的相關算法可以分為多目圖像深度估計和單目圖像深度估計兩大類。其中，單目深度估計算法有著更廣闊的應用前景，在單目圖像深度估計領域中，主要通過訓練網絡模型的方式實現深度信息估計。

相關技術中，在訓練用于進行深度信息估計的網絡模型時，往往需要獲取到足夠多的訓練樣本，所需要的訓練樣本形式為：單目圖像和配套的單通道深度譜，深度譜上每個位置的取值代表原圖對應位置與相機的距離。一般的獲取方式為通過專門的傳感器采集圖像，以獲取到具有深度信息的圖像。但是該獲取方式獲取成本較高，且獲取效率較低，無法用于快速獲取到具有深度信息的圖像。

發明內容

本申請實施例提供了一種視差數據生成方法、裝置、計算機設備及存儲介質。所述技術方案如下：

一方面，本申請實施例提供了視差數據生成方法，所述方法包括：

獲取至少一組原始圖像對，所述原始圖像對中包含原始左目圖像和原始右目圖像，所述原始左目圖像和所述原始右目圖像是在左目視角和右目視角下觀察同一場景所對應的圖像；

對所述原始圖像對進行光流提取，得到各組所述原始圖像對對應的光流值；

基于所述光流值對所述原始圖像對進行光流篩選，得到目標單目圖像和所述目標單目圖像中各個像素點對應的目標光流值，其中，光流篩選后各個像素點對應目標光流值之間的相對大小關系與視差數據之間的相對大小關系相同，所述目標單目圖像為原始左目圖像或原始右目圖像；

基于所述目標光流值，生成所述目標單目圖像對應的目標視差數據。

另一方面，本申請實施例提供了一種一種視差數據生成裝置，所述裝置包括：

獲取模塊，用于獲取至少一組原始圖像對，所述原始圖像對中包含原始左目圖像和原始右目圖像，所述原始左目圖像和所述原始右目圖像是在左目視角和右目視角下觀察同一場景所對應的圖像；

光流提取模塊，用于對所述原始圖像對進行光流提取，得到各組所述原始圖像對對應的光流值；

光流篩選模塊，用于基于所述光流值對所述原始圖像對進行光流篩選，得到目標單目圖像和所述目標單目圖像中各個像素點對應的目標光流值，其中，光流篩選后各個像素點對應目標光流值之間的相對大小關系與視差數據之間的相對大小關系相同，所述目標單目圖像為原始左目圖像或原始右目圖像；

生成模塊，用于基于所述目標光流值，生成所述目標單目圖像對應的目標視差數據。

另一方面，本申請實施例提供了一種計算機設備，所述計算機設備包括處理器和存儲器，所述存儲器中存儲有至少一條程序代碼，所述程序代碼由所述處理器加載并執行以實現如上述方面所述的視差數據生成方法。