HITNet: Hierarchical Iterative Tile Refinement Network for Real-time Stereo Matching

This paper presents HITNet, a novel neural network architecture for real-time stereo matching. Contrary to many recent neural network approaches that operate on a full cost volume and rely on 3D convolutions, our approach does not explicitly build a volume and instead relies on a fast multi-resolution initialization step, differentiable 2D geometric propagation and warping mechanisms to infer disparity hypotheses. To achieve a high level of accuracy, our network not only geometrically reasons about disparities but also infers slanted plane hypotheses allowing to more accurately perform geometric warping and upsampling operations. Our architecture is inherently multi-resolution allowing the propagation of information across different levels. Multiple experiments prove the effectiveness of the proposed approach at a fraction of the computation required by state-of-the-art methods. At the time of writing, HITNet ranks 1st-3rd on all the metrics published on the ETH3D website for two view stereo, ranks 1st on most of the metrics among all the end-to-end learning approaches on Middlebury-v3, ranks 1st on the popular KITTI 2012 and 2015 benchmarks among the published methods faster than 100ms.

PDF Abstract CVPR 2021 PDF CVPR 2021 Abstract

Datasets


Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Stereo Depth Estimation KITTI2015 HITNET three pixel error 2.43 # 1
Stereo Disparity Estimation Scene Flow HITNet XL EPE 0.36 # 7
one pixel error 4.09 # 1
three pixel error 2.21 # 1
Stereo Disparity Estimation Scene Flow HITNet L EPE 0.43 # 6
one pixel error 4.70 # 2
three pixel error 2.57 # 2
Stereo Disparity Estimation Scene Flow HITNet EPE 0.529 # 5
one pixel error 5.52 # 3
three pixel error 3.00 # 4

Methods