Skip to content

kovibalu/saw_release

master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Shading Annotations in the Wild (SAW)

Code and data for paper "Shading Annotations in the Wild".

Installation

Dependencies

Our code was tested on Ubuntu 14.04. As a first step, clone our repo:

git clone https://github.com/kovibalu/saw_release.git

Then install the python dependencies by running:

sudo ./install/install_python.sh

If you would like to run our trained model, you will need to install Caffe. We slightly modified the implementation of Bansal et. al for our purposes. To check out our Caffe version which is included as a submodule, run:

git submodule update --init --recursive

Then build Caffe after editing the Makefile.config depending on your configuration with:

cd caffe
make all -j
make pycaffe -j

Download Data

To download all data related to the dataset, run:

./download_saw.sh

The whole dataset download size is ~28.0GB, please see the documentation in the script for a detailed breakdown of sizes for the different parts of the dataset. For detailed documentation on the format of the downloaded annotations in saw/saw_annotations_json see ANNO_FORMAT.md.

Usage

Precision-recall Curves

To generate the precision-recall curves in our paper for all baselines and our method, run:

python main.py generate_pr

You can select which baselines to evaluate in main.py.

Generating Pixel Labels

To generate the pixel labels from the SAW annotations and NYUv2 depth dataset depth and normal maps, run:

python main.py generate_labels

Citation

Please cite our paper if you use our code or data:

@article{kovacs17shading,
	author = "Balazs Kovacs and Sean Bell and Noah Snavely and Kavita Bala",
	title = "Shading Annotations in the Wild",
	journal = "Computer Vision and Pattern Recognition (CVPR)",
	year = "2017",
}

Contact

Please contact Balazs Kovacs with any questions.

About

Code and data for Shading Annotations in the Wild

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published