/Structure-Guided-Ranking-Loss

Structure-Guided Ranking Loss for Single Image Depth Prediction

Primary LanguagePythonOtherNOASSERTION

Structure-Guided Ranking Loss for Single Image Depth Prediction

This repository contains a pytorch implementation of our CVPR2020 paper "Structure-Guided Ranking Loss for Single Image Depth Prediction". Project Page Teaser Image

Changelog

  • [Jun. 2020] Initial release

To do

  • Mix data training

Prerequisites

  • 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

Inference

# 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}
}

Dataset

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.

Lisence

Research only