Awesome
PyTorch-HITNet-Hierarchical-Iterative-Tile-Refinement-Network-for-Real-time-Stereo-Matching
HITNet implementation using PyTorch
This is a repository including code implementing Google paper HITNet: Hierarchical Iterative Tile Refinement Network for Real-time Stereo Matching
This project is an initial version, which can train and test the model but may contain some errors and need further modification and debugging. If you find any issue about my code, please open issues or contact me (mjitglv@gmail.com) as soon as possible.
Currently this project cannot reproduce the accuracy and speed reported in the original paper. In terms of the speed, the official implementation uses their optimized cuda op to accelerate the reference and training.(Please refer to their official repository, which has not inculded the model code yet).
Thanks for the help of Vladimir Tankovich, who has proposed this great stereo network with his team and provided me with a lot of details and clarifications of the original paper.
Also, I would thank @xy-guo, who proposed the amazing GwcNet, since the code was partially borrowed from his repository.
Requirements
Pytorch = 1.1 Python = 3.6.8 CUDA10
Slant Ground-truth
Slant parameter GT is here, which is generated using least square fit and RANSAC. The code for slant parameter generation is here.