arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.07094cs.CVcs.RO
收藏

PVStereo:用于端到端自监督立体匹配的金字塔投票模块

PVStereo: Pyramid Voting Module for End-to-End Self-Supervised Stereo Matching

Hong Kong University of Science and Technology(香港科技大学);University of California San Diego(加利福尼亚大学圣迭戈分校)

Hengli Wang, Rui Fan, Peide Cai, Ming Liu

摘要:

使用深度卷积神经网络(DCNN)的监督学习在立体匹配中得到了广泛采用。然而,获取具有良好标记的地面真值的大规模数据集是繁琐且劳动密集型的,使得基于监督学习的方法通常难以在实践中实现。为了克服这个缺点,我们提出了一个强大的和有效的自监督立体匹配方法,包括一个金字塔投票模块(PVM)和一个新的DCNN架构,称为OptStereo。具体来说,我们的OptStereo首先构建多尺度成本卷,然后采用循环单元以高分辨率迭代更新视差估计;而我们的PVM可以生成可靠的半密集视差图像,可以用来监督OptStereo训练。此外,我们发布了HKUST-Drive数据集,这是一个大规模的合成立体数据集,在不同的光照和天气条件下收集,用于研究目的。大量的实验结果表明,我们的自监督立体匹配方法的有效性和效率的KITTI立体基准和我们的HKUST-Drive数据集。PVStereo是我们性能最好的实现,大大优于所有其他最先进的自监督立体匹配方法。我们的项目页面可在sites.google.com/view/pvstereo上找到。

英文摘要:

Supervised learning with deep convolutional neural networks (DCNNs) has seen huge adoption in stereo matching. However, the acquisition of large-scale datasets with well-labeled ground truth is cumbersome and labor-intensive, making supervised learning-based approaches often hard to implement in practice. To overcome this drawback, we propose a robust and effective self-supervised stereo matching approach, consisting of a pyramid voting module (PVM) and a novel DCNN architecture, referred to as OptStereo. Specifically, our OptStereo first builds multi-scale cost volumes, and then adopts a recurrent unit to iteratively update disparity estimations at high resolution; while our PVM can generate reliable semi-dense disparity images, which can be employed to supervise OptStereo training. Furthermore, we publish the HKUST-Drive dataset, a large-scale synthetic stereo dataset, collected under different illumination and weather conditions for research purposes. Extensive experimental results demonstrate the effectiveness and efficiency of our self-supervised stereo matching approach on the KITTI Stereo benchmarks and our HKUST-Drive dataset. PVStereo, our best-performing implementation, greatly outperforms all other state-of-the-art self-supervised stereo matching approaches. Our project page is available at sites.google.com/view/pvstereo.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑