【论文学习笔记-14】Bilateral Grid Learning for Stereo Matching Networks
Main Contribution
提出了一个基于双边网格的保边的cost volume上采样,主要的计算在低分辨率下进行,可以保持高速度,同时精度高。
Method
关键模块是Cost volume Upsampling in Bilateral Grid(CUBG)module,该模块输入一个低分辨率的Cost volume,先经过一个四维的双边滤波,然后通过Feature map的2d切片对每个通道进行guide并上采样为原分辨率。

本文将该模块加入了GCNet,PSMNet,GANet和DeepPrunerFast的CostVolume中对低分辨率Cost Volume进行上采样,其它模块不变。
本文基于CUBG提出了BGNet

特征提取:前三层ks = 3*3,stride为2,1,1的卷积;然后接四个残差快,stride为1,2,2,1。得到1/8分辨率;之后接一个沙漏结构获扩大感受野。最后所有1/8分辨率的feature拼在一起,得到352通道的特征。
Cost Aggregation
使用group-wise correlation,将特征分为N组,每组之间算内积相似度,N=44(组),最后得到BDNHW的costvolume。随后通过一个沙漏结构和两层3D卷积降低N的数量:44->16。之后通过一个U

本文提出了一种新的Cost Volume上采样方法——CUBG模块,它利用双边滤波在低分辨率下进行计算,以实现高速且高精度的立体匹配。CUBG被整合到GCNet、PSMNet、GANet和DeepPruner Fast的Cost Volume中,形成新的BGNet网络。特征提取部分包括多层卷积和残差块,最后通过3D卷积和沙漏结构进行Cost Aggregation。实验表明,这种方法在保持效率的同时提高了匹配精度,损失函数采用了L1 loss。
最低0.47元/天 解锁文章
776

被折叠的 条评论
为什么被折叠?



