【论文学习笔记-14】Bilateral Grid Learning for Stereo Matching Networks

【论文学习笔记-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-Net结构的3D卷积网络进行Cost Aggregation。得到低分辨率的Cost Volume。

最后原分辨率的Cost Volume通过Soft argmin获得视差预测图。loss为L1 loss。

Experiment

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值