論文 | DISN三維重建模型《DISN: Deep Implicit Surface Network for High-quality Single-view 3D Reconstruction》

一 寫在前面

未經(jīng)允許,不得轉(zhuǎn)載鸠补,謝謝~~

這篇文章主要講的是進行三維重建的過程膘滨,與前面提到的這篇文章最大的不同之處在于其表示方式既不是基于點的voxel或者point clouds耀鸦,也不是基于網(wǎng)格的mesh寝衫。

二 主要介紹

2.1 背景介紹

  1. 目前三維重建方向已經(jīng)很受關(guān)注顷扩,尤其是single view image reconstruction。

目前深度學習已經(jīng)展現(xiàn)了在這個任務上的優(yōu)越性慰毅。

  1. 在現(xiàn)有的方法中:

point based method (voxel, point cloud)

  • 優(yōu)點:易于使用CNN(歐幾里得表示)
  • 缺點:分辨率很受限

mesh based method

  • 優(yōu)點:表示更精細
  • 缺點:受限于固定的拓撲結(jié)構(gòu)
  1. 而且在以上兩種point based與mesh based的方法中大多只能使用Chamfer distance(CD)Earth Mover Distance(EMD)來度量預測模型與gt模型之間的距離隘截,而這兩種距離本身只能近似估計。

2.2 文章方法概述

我覺得文章主要可以歸為以下兩大貢獻:

  • 引入SDF(signed distance function)來表示3D物體汹胃。
  • 提出DISN(deep implicit surface network)來預測SDF婶芭;

另外其實驗結(jié)果確實還不錯,應該是第一個能把物體的孔洞等細節(jié)都恢復出來的模型:

三 文章方法

3.1 SDF (signed distance function)表示方法

  • SDF是一個將3D物體中的點p=(x,y,z)映射到實數(shù)s=SDF(p)的連續(xù)函數(shù)着饥。
  • 如下圖所示:


  • s 的絕對值表示這個點到物體表面的距離
  • s=0: 點在物體表面
  • s>0:點在物體外部
  • s<0:點在物體內(nèi)部
    這樣就可以把3D模型表示出來雕擂。

3.2 DISN(deep implicit surface network)網(wǎng)絡(luò)結(jié)構(gòu)

DISN用圖像作為輸入,然后可以預測任意一個給定點p(x,y,z)的SDF值贱勃。這樣表示DISN的預測模型的分辨率是完全不受限制的井赌,因為我們可以給任意多的點。

結(jié)構(gòu)圖如下:


DISN主要分成兩個部分:

  • camer pose estimation :評估將3D物體投影到這張2D圖像的相機參數(shù)贵扰;
  • SDF prediction:預測給定點的SDF值仇穗。

參照網(wǎng)絡(luò)結(jié)構(gòu)圖,主要包含以下幾個主要步驟:

  1. 輸入:input image戚绕,point p(x,y,z)
  2. 對input image進行相機參數(shù)的評估纹坐;
  3. 用CNN encoder得到輸入圖像的global feature即各層的feature map;
  4. 用預測到的相機參數(shù)將點p映射到2D image plane舞丛,然后進行這個點的局部特征提取
  5. 最后將全局特征耘子、局部特征依次與點特征結(jié)合再進行特征提取,得到兩路SDF值球切,最后做個融合谷誓。

接下來介紹一些主要步驟中比較重要的部分:

1 相機參數(shù)評估

  • 相機參數(shù)評估是一個獨立的部分,也是用深度卷積神經(jīng)網(wǎng)絡(luò)來預測圖像的相機參數(shù)吨凑。
  • 其意思可以理解將一個3D物體投影到這張?zhí)囟ǖ妮斎雸D像上所需要的相機參數(shù)是什么樣子的捍歪。
  • 估計出來的相機參數(shù)在后面對每個點求局部特征的時候會被用到。
  • 其結(jié)構(gòu)圖:

用的是ShapeNet數(shù)據(jù)集進行這個網(wǎng)絡(luò)的訓練鸵钝。PC表示點云糙臼,GT Cam Pred Cam分別表示ground truth的相機參數(shù),預測出來的相機參數(shù)恩商。大概過程就是對輸入進行一定的旋轉(zhuǎn)等操作得到一個預測的相機參數(shù)变逃,然后用預測的相機參數(shù)下的點分布與gt下的點位置分布去做一個比較,得到一個loss

(但這個具體的過程涉及坐標系轉(zhuǎn)換怠堪、旋轉(zhuǎn)等我還沒有弄懂揽乱。)

2 局部特征提取

這個想法其實也挺簡單的末患,先看一下圖吧:


  • 首先用已知的相機參數(shù)(訓練)或者預測得到的相機參數(shù)(測試)將點p投影回二維的圖像平面;
  • 之前在求圖像的全局特征的時候就已經(jīng)獲取到了各層的feature map锤窑;
  • 或者這個點對應各個feature map中的特征(具體操作的時候為了包含保證各個大小一致璧针,對小的feature map用雙線性差值作了尺寸恢復)
  • 把每個feature map提取到的局部特征(圖中紅色的小方塊)直接做一個連接就構(gòu)成了我們想要的local feature。
  • 文章實驗證明這個local feature能幫助解決細節(jié)(孔渊啰、細小的地方)恢復的問題探橱。(a和b分別是沒有局部特征和有局部特征的效果)


3 SDF預測及損失函數(shù)

  • 原本的ShapeNet數(shù)據(jù)集提供的是物體的3D mesh結(jié)構(gòu)
  • 為每個物體的3D mesh生成256^3的SDF作為ground truth: SDF(I)
  • 這樣在與預測得到的SDF的時候就不需要用近似的CD、EMD距離來求loss绘证,直接用精確對應的點點關(guān)系進行計算即可隧膏。
  • 然后在loss設(shè)計的過程中考慮到模型會更加關(guān)注接近表面的點,所以最終的loss函數(shù):
  • m表示權(quán)重系數(shù)嚷那,如果在一定范圍內(nèi)則權(quán)重為m1, 否則為m2
  • 為了讓表面的點更有權(quán)重胞枕,m1 > m2 ;
  • f(I,p)表示DISN網(wǎng)絡(luò)對輸入圖像I,對給定點p預測得到的SDF值魏宽;
  • SDF(I)(p)表示點p的SDF的ground truth值腐泻;
  • 距離用的是L1距離;

四 還要再學習的幾個地方

  1. 相機參數(shù)估計網(wǎng)絡(luò)
  • E.InsafutdinovandA.Dosovitskiy. Unsupervisedlearningofshapeandposewithdifferentiable point clouds. In NeurIPS, 2018.
  • Y. Zhou, C. Barnes, J. Lu, J. Yang, and H. Li. On the continuity of rotation representations in neural networks. arXiv preprint arXiv:1812.07035, 2018.
  • 本文改進的方法
  1. 如何獲取輸入圖像I的SDF ground truth
  • F. S. Sin, D. Schroeder, and J. Barbiˇc. Vega: non-linear fem deformable object simulator. In Computer Graphics Forum, 2013.
  • H. Xu and J. Barbiˇc. Signed distance ?elds for polygon soup meshes. In Proceedings of Graphics Interface 2014, pages 35–41. Canadian Information Processing Society, 2014.
  1. 獲取到物體的預測出來的SDF信息后队询,如何進行mesh的重建工作
  • W. E. Lorensen and H. E. Cline. Marching cubes: A high resolution 3d surface construction algorithm. In ACM siggraph computer graphics, 1987.
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末派桩,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子蚌斩,更是在濱河造成了極大的恐慌铆惑,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,036評論 6 506
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件送膳,死亡現(xiàn)場離奇詭異员魏,居然都是意外死亡,警方通過查閱死者的電腦和手機叠聋,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,046評論 3 395
  • 文/潘曉璐 我一進店門撕阎,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人晒奕,你說我怎么就攤上這事闻书。” “怎么了脑慧?”我有些...
    開封第一講書人閱讀 164,411評論 0 354
  • 文/不壞的土叔 我叫張陵,是天一觀的道長砰盐。 經(jīng)常有香客問我闷袒,道長,這世上最難降的妖魔是什么岩梳? 我笑而不...
    開封第一講書人閱讀 58,622評論 1 293
  • 正文 為了忘掉前任囊骤,我火速辦了婚禮晃择,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘也物。我一直安慰自己宫屠,他們只是感情好,可當我...
    茶點故事閱讀 67,661評論 6 392
  • 文/花漫 我一把揭開白布滑蚯。 她就那樣靜靜地躺著浪蹂,像睡著了一般。 火紅的嫁衣襯著肌膚如雪告材。 梳的紋絲不亂的頭發(fā)上坤次,一...
    開封第一講書人閱讀 51,521評論 1 304
  • 那天,我揣著相機與錄音斥赋,去河邊找鬼缰猴。 笑死,一個胖子當著我的面吹牛疤剑,可吹牛的內(nèi)容都是我干的滑绒。 我是一名探鬼主播,決...
    沈念sama閱讀 40,288評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼隘膘,長吁一口氣:“原來是場噩夢啊……” “哼蹬挤!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起棘幸,我...
    開封第一講書人閱讀 39,200評論 0 276
  • 序言:老撾萬榮一對情侶失蹤焰扳,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后误续,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體吨悍,經(jīng)...
    沈念sama閱讀 45,644評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,837評論 3 336
  • 正文 我和宋清朗相戀三年蹋嵌,在試婚紗的時候發(fā)現(xiàn)自己被綠了育瓜。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 39,953評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡栽烂,死狀恐怖躏仇,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情腺办,我是刑警寧澤焰手,帶...
    沈念sama閱讀 35,673評論 5 346
  • 正文 年R本政府宣布,位于F島的核電站怀喉,受9級特大地震影響书妻,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜躬拢,卻給世界環(huán)境...
    茶點故事閱讀 41,281評論 3 329
  • 文/蒙蒙 一躲履、第九天 我趴在偏房一處隱蔽的房頂上張望见间。 院中可真熱鬧,春花似錦工猜、人聲如沸米诉。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,889評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽史侣。三九已至,卻和暖如春犹褒,著一層夾襖步出監(jiān)牢的瞬間抵窒,已是汗流浹背叠骑。 一陣腳步聲響...
    開封第一講書人閱讀 33,011評論 1 269
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留掉房,地道東北人慰丛。 一個月前我還...
    沈念sama閱讀 48,119評論 3 370
  • 正文 我出身青樓诅病,卻偏偏與公主長得像贤笆,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子篡殷,可洞房花燭夜當晚...
    茶點故事閱讀 44,901評論 2 355

推薦閱讀更多精彩內(nèi)容