PSMNet代码已提供。

简介：
Pyramid Stereo Matching Network (PSMNet) 的代码。最近的研究表明，从立体图像对中进行深度估计可以被表述为一个监督学习任务，并利用卷积神经网络 (CNNs) 来解决。然而，现有的架构依赖于基于区域的Siamese网络，这些网络缺乏充分利用上下文信息来在图像失配区域中寻找对应点的能力。为了解决这一问题，我们提出PSMNet，这是一种基于金字塔结构的立体匹配网络，它包含两个主要模块：空间金字塔池化和3D CNN。空间金字塔池化模块通过聚合不同尺度和位置的上下文信息，从而充分利用全局上下文信息的强大能力，构建一个代价体积。3D CNN 学习使用堆叠的多个 hourglass 网络与中间监督来规范化代价体积。我们提出的方法在多个基准数据集上进行了评估。在2018年3月18日之前，我们的方法在KITTI 2012和2015排行榜上名列第一。