Code and data for Shading Annotations in the Wild
Python Shell
Switch branches/tags
Nothing to show
Latest commit d36b62a Apr 8, 2017 @kovibalu Updated license
Failed to load latest commit information.
caffe @ a871af4
.gitignore Added gitignore Mar 31, 2017
.gitmodules Using the MarrRevisited caffe branch Mar 31, 2017 Fixed spacing Mar 31, 2017
LICENSE Updated license Apr 8, 2017 Added saw_heatmaps Apr 8, 2017 Initial commit Mar 31, 2017 Fix Apr 2, 2017 Initial commit Mar 31, 2017 Plot last curve with black Apr 4, 2017

Shading Annotations in the Wild (SAW)

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



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

git clone

Then install the python dependencies by running:

sudo ./install/

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:


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


Precision-recall Curves

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

python generate_pr

You can select which baselines to evaluate in

Generating Pixel Labels

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

python generate_labels


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

	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",


Please contact Balazs Kovacs with any questions.