以非局部的解碼器-擠壓-激勵網路及自適應深度列表達成

基於編碼器-解碼器的單鏡頭深度估計任務

摘要:

單鏡頭深度估計是計算機視覺中的一個重要議題。近年來, 基於卷積神經網路的編碼器-解碼器架構中展現了合理的結果。在一個強大的編碼器下,人們發現即使是簡 單的上採樣過程也能達到良好的準確度。然而,為了達到更高質量的深度估計,解碼器也是需要改進的地方 。即便現在很少有合理且確切有用的方法對上採樣過程做出貢獻。因此,我們提出了兩個新穎的單鏡頭深度 估計網路模組設計,兩個模組皆從全局的角度來改善上採樣過程。第一個為非局部解碼器-擠壓-激勵網路模組, 該模組用於解碼中間過程的數值校準。第二個為自適應深度列表,用於進行整體深度圖數值分布的調整。結合上 述提出的技術,我們的方法在NYU Depth V2深度資料集上進行了驗證,並在深度準確度上達到目前非常好的做法。

 

NYU Depth V2深度資料集樣本

 

 

架構設計:

我們提出的方法的概念圖如圖二所示,概念圖可以簡單理解為 我們所設計的模組共同去改善編碼器-解碼器的網路架構以達到更好的深度估計準確度。而我們提出的非局部的 解碼器-擠壓-激勵模組校準位於解碼器的下方,該模組於解碼過程中去校準不同解析度的特徵圖,以達到改善中 間過程的數值分布。而提出的第二個模組的自適應深度列表在模型的後端,該模組位於的輸出端來有效的調整輸 出數值的範圍。我們依靠上面兩個模組來有效的改善深度估計的準確度。

 

網路架構示意圖

 

 

 

 

Made by  萬偉中