Skip to content
Code and data for Shading Annotations in the Wild
Python Shell
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.
caffe @ a871af4

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.

You can’t perform that action at this time.