/polyrnn-pp

Code for Polygon-RNN++ (CVPR 2018)

Primary LanguageJupyter NotebookGNU General Public License v3.0GPL-3.0

PolygonRNN++

This is the official repository for the Polygon-RNN++ project (CVPR-2018). For technical details, please refer to:

Efficient Interactive Annotation of Segmentation Datasets with Polygon-RNN++
David Acuna*, Huan Ling*, Amlan Kar*, Sanja Fidler (* denotes equal contribution)
CVPR 2018
[Paper] [Video] [Project Page]
Model

Usage

  1. Clone the repository
git clone https://github.com/davidjesusacu/polyrnn && cd polyrnn
  1. Install dependencies
    (Note: Using a GPU (and tensorflow-gpu) is recommended. The model will run on a CPU, albeit slowly.)
virtualenv env
source env/bin/activate
pip install -r requirements.txt
  1. Download the pre-trained models and graphs (448 MB)
    (These models were trained on the Cityscapes Dataset)
./models/download_and_unpack.sh 
  1. Run demo_inference.sh
./src/demo_inference.sh 

This should produce results in the output/ folder that look like ex2 ex1

Walkthrough

Checkout the ipython notebook that provides a simple walkthrough demonstrating how to run our model on sample input image crops

Coming Soon

  • Online Demo
  • Training Code

If you use this code, please cite:

@inproceedings{AcunaCVPR18,
title={Efficient Interactive Annotation of Segmentation Datasets with Polygon-RNN++},
author={David Acuna and Huan Ling and Amlan Kar and Sanja Fidler},
booktitle={CVPR},
year={2018}
}

@inproceedings{CastrejonCVPR17,
title = {Annotating Object Instances with a Polygon-RNN},
author = {Lluis Castrejon and Kaustav Kundu and Raquel Urtasun and Sanja Fidler},
booktitle = {CVPR},
year = {2017}
}