Skip to content
Code for the CVPR 2019 paper "Learning Single-Image Depth from Videos using Quality Assessment Networks"
Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
src first commit Jul 4, 2019
LICENSE first commit Jul 4, 2019
README.md Update README.md Jul 13, 2019

README.md

Learning Single-Image Depth from Videos using Quality Assessment Networks

Code for reproducing the results in the following paper:

Learning Single-Image Depth from Videos using Quality Assessment Networks
Weifeng Chen, Shengyi Qian, Jia Deng
Conference on Computer Vision and Pattern Recognition (CVPR), 2019.

Please check the project site for more details.

Example outputs on the Depth in the Wild (DIW) test set

qual_outputs

Setup

  1. The code is written in python 2.7.13, using pytorch 0.2.0_4. Please make sure that you install the correct pytorch version as later versions may cause the code to break.

  2. Clone this repo.

git clone git@github.com:princeton-vl/YouTube3D.git
  1. Download data_model.tar.gz into path YouTube3D, then untar:
cd YouTube3D
tar -xzvf data_model.tar.gz
  1. Download and unpack the images from Depth in the Wild dataset. Edit DIW_test.csv under YouTube3D/data so that all the image paths are absolute paths.

Evaluating the pretrained models

To evaluate the pre-trained model EncDecResNet trained on ImageNet + ReDWeb + DIW + YouTube3D on the DIW dataset, run the following command:

cd YouTube3D/src 
python test.py -t DIW_test.csv -model exp/YTmixReD_dadlr1e-4_DIW_ReDWebNet_1e-6_bs4/models/model_iter_753000.bin

In case you want to get the qualitative outputs, append a -vis flag and the qualitative outputs will be in the folder visualize:

mkdir visualize
python test.py -t DIW_test.csv -model exp/YTmixReD_dadlr1e-4_DIW_ReDWebNet_1e-6_bs4/models/model_iter_753000.bin -vis

Contact

Please send any questions or comments to Weifeng Chen at wfchen@umich.edu.

You can’t perform that action at this time.