A New Distributional Ranking Loss With Uncertainty: Illustrated in Relative Depth Estimation


Creative Commons License

Mertan A., Şahin Y. H., Duff D. J., Ünal G.

8th International Conference on 3D Vision (3DV), ELECTR NETWORK, 25 - 28 Kasım 2020, ss.1079-1088 identifier identifier

  • Yayın Türü: Bildiri / Tam Metin Bildiri
  • Doi Numarası: 10.1109/3dv50981.2020.00118
  • Basıldığı Ülke: ELECTR NETWORK
  • Sayfa Sayıları: ss.1079-1088
  • İstanbul Teknik Üniversitesi Adresli: Evet

Özet

We propose a new approach for the problem of relative depth estimation from a single image. Instead of directly regressing over depth scores, we formulate the problem as estimation of a probability distribution over depth and aim to learn the parameters of the distributions which maximize the likelihood of the given data. To train our model, we propose a new ranking loss, Distributional Loss, which tries to increase the probability of farther pixel's depth being greater than the closer pixel's depth. Our proposed approach allows our model to output confidence in its estimation in the form of standard deviation of the distribution. We achieve state of the art results against a number of baselines while providing confidence in our estimations. Our analysis show that estimated confidence is actually a good indicator of accuracy. We investigate the usage of confidence information in a downstream task of metric depth estimation, to increase its performance.