This repository contains a pytorch implementation of our CVPR2020 paper "Structure-Guided Ranking Loss for Single Image Depth Prediction".
Project Page
- [Jun. 2020] Initial release
- Mix data training
- Pytorch >= 0.4.1
- CUDA >= 0.8
- Python >= 2.7
- glob, matplotlib
- Need to compile the syncbn module in models/syncbn. Note that the directory of the syncbn module should be modified in some .py files (i.e., DepthNet.py, resnet.py and networks.py)
- Download the model.pth.tar
# Before running, you should set the CUDA_VISIBLE_DEVICES in demo.sh
bash demo.sh
If you find our work useful in your research, please consider citing the paper.
@InProceedings{Xian_2020_CVPR,
author = {Xian, Ke and Zhang, Jianming and Wang, Oliver and Mai, Long and Lin, Zhe and Cao, Zhiguo},
title = {Structure-Guided Ranking Loss for Single Image Depth Prediction},
booktitle = {The IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2020}
}
Our HRWSI dataset is for research only! Some researchers may interested in the stereo data, so we provide the right views here. Please let me know if you have any questions.
Research only