/cascade-stereo

cascade-stereo

Primary LanguagePythonMIT LicenseMIT

Casacde-Stereo(CVPR2020 Oral)

Official source code of paper Cascade Cost Volume for High-Resolution Multi-View Stereo and Stereo Matching, arxiv

Installation

Requirements

  • python 3.6
  • Pytorch == 1.2
  • CUDA >= 9.0
pip install -r requirements.txt  --user

Option: Apex

install apex to enable synchronized batch normalization

git clone https://github.com/NVIDIA/apex.git
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

CasMVSNet(Multi View Stereo)

Refer to MVS-README.md

CasStereoNet(Stereo Matching)

Refer to Stereo-Matching-README.md

Citation

If you find this code useful in your research, please cite:

@inproceedings{gu2019cas,
  title={Cascade Cost Volume for High-Resolution Multi-View Stereo and Stereo Matching},
  author={Gu, Xiaodong and Fan, Zhiwen and Zhu, Siyu and Dai, Zuozhuo and Tan, Feitong and Tan, Ping},
  journal={arxiv preprint arXiv:1912.06378},
  year={2019}
}

Acknowledgements

Thanks to Xiaoyang Guo for opening source of his excellent works GwcNet and MVSNet-pytorch. Thanks to Yao Yao for opening source of his excellent work MVSNet.