一种基于空间上下文注意力网络的立体匹配方法

摘要

立体匹配可以看作是有监督的学习任务,通过将大量的左右图像输入到卷积神经网络中进行训练,可以获得性能良好的视差图。但是,当前的网络结构对病态区域的视差估计仍然存在困难。为了解决这个问题,提出了一种空间上下文注意力网络,这个网络可以通过聚合全局上下文信息来提高对病态区域的视差估计准确性。实验结果表明,提出的方法在合成数据集Scene Flow和真实场景数据集KITTI 2015中对病态区域的视差估计得了较好的效果。