References & Citations
Computer Science > Computer Vision and Pattern Recognition
Title: Wasserstein Distances for Stereo Disparity Estimation
(Submitted on 6 Jul 2020 (v1), last revised 29 Mar 2021 (this version, v2))
Abstract: Existing approaches to depth or disparity estimation output a distribution over a set of pre-defined discrete values. This leads to inaccurate results when the true depth or disparity does not match any of these values. The fact that this distribution is usually learned indirectly through a regression loss causes further problems in ambiguous regions around object boundaries. We address these issues using a new neural network architecture that is capable of outputting arbitrary depth values, and a new loss function that is derived from the Wasserstein distance between the true and the predicted distributions. We validate our approach on a variety of tasks, including stereo disparity and depth estimation, and the downstream 3D object detection. Our approach drastically reduces the error in ambiguous regions, especially around object boundaries that greatly affect the localization of objects in 3D, achieving the state-of-the-art in 3D object detection for autonomous driving. Our code will be available at this https URL
Submission history
From: Divyansh Garg [view email][v1] Mon, 6 Jul 2020 21:37:50 GMT (5722kb,D)
[v2] Mon, 29 Mar 2021 09:42:37 GMT (19863kb,D)
Link back to: arXiv, form interface, contact.