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
Permalink
Failed to load latest commit information.
caffe @ a871af4
install
.gitignore Added gitignore Mar 31, 2017
.gitmodules Using the MarrRevisited caffe branch Mar 31, 2017
ANNO_FORMAT.md Fixed spacing Mar 31, 2017
LICENSE Updated license Apr 8, 2017
README.md
constants.py
download_saw.sh Added saw_heatmaps Apr 8, 2017
evaluation.py Initial commit Mar 31, 2017
label_generation.py Fix Apr 2, 2017
main.py
metrics.py
net_tools.py Initial commit Mar 31, 2017
utils.py Plot last curve with black Apr 4, 2017

README.md

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.