1. SMD-Nets: Stereo Mixture Density Networks
- Author
-
Carolin Schmitt, Fabio Tosi, Yiyi Liao, and Andreas Geiger
- Subjects
FOS: Computer and information sciences ,Flexibility (engineering) ,Computer science ,business.industry ,Computer Vision and Pattern Recognition (cs.CV) ,Deep learning ,Computer Science - Computer Vision and Pattern Recognition ,ComputingMethodologies_IMAGEPROCESSINGANDCOMPUTERVISION ,020206 networking & telecommunications ,02 engineering and technology ,Solid modeling ,Classification of discontinuities ,Pattern recognition (psychology) ,0202 electrical engineering, electronic engineering, information engineering ,Mixture distribution ,020201 artificial intelligence & image processing ,Artificial intelligence ,Representation (mathematics) ,business ,Image resolution ,Algorithm - Abstract
Despite stereo matching accuracy has greatly improved by deep learning in the last few years, recovering sharp boundaries and high-resolution outputs efficiently remains challenging. In this paper, we propose Stereo Mixture Density Networks (SMD-Nets), a simple yet effective learning framework compatible with a wide class of 2D and 3D architectures which ameliorates both issues. Specifically, we exploit bimodal mixture densities as output representation and show that this allows for sharp and precise disparity estimates near discontinuities while explicitly modeling the aleatoric uncertainty inherent in the observations. Moreover, we formulate disparity estimation as a continuous problem in the image domain, allowing our model to query disparities at arbitrary spatial precision. We carry out comprehensive experiments on a new high-resolution and highly realistic synthetic stereo dataset, consisting of stereo pairs at 8Mpx resolution, as well as on real-world stereo datasets. Our experiments demonstrate increased depth accuracy near object boundaries and prediction of ultra high-resolution disparity maps on standard GPUs. We demonstrate the flexibility of our technique by improving the performance of a variety of stereo backbones., Comment: CVPR 2021. Project Page: https://github.com/fabiotosi92/SMD-Nets
- Published
- 2021
- Full Text
- View/download PDF