Annotating Object Instances with a Polygon-RNN

Lluis Castrejon, Kaustav Kundu, Raquel Urtasun, Sanja Fidler
4/18/2017
cs.CV

Official code: fidler-lab/polyrnn-pp-pytorch (700 stars, PyTorch, last commit 2019-07-24).

Abstract

We propose an approach for semi-automatic annotation of object instances. While most current methods treat object segmentation as a pixel-labeling problem, we here cast it as a polygon prediction task, mimicking how most current datasets have been annotated. In particular, our approach takes as input an image crop and sequentially produces vertices of the polygon outlining the object. This allows a human annotator to interfere at any time and correct a vertex if needed, producing as accurate segmentation as desired by the annotator. We show that our approach speeds up the annotation process by a factor of 4.7 across all classes in Cityscapes, while achieving 78.4% agreement in IoU with original ground-truth, matching the typical agreement between human annotators. For cars, our speed-up factor is 7.3 for an agreement of 82.2%. We further show generalization capabilities of our approach to unseen datasets.

View on arXivView PDF

Code Implementations(1)

PyTorch

PyTorch training/tool code for Polygon-RNN++ (CVPR 2018)

700106Aug 14, 20187 years ago
annotationcvpr2018deep-learninginstance-segmentationlabelling+2 more

Similar Papers

Cite this paper

@article{castrejon2017annotating,
  title  = {Annotating Object Instances with a Polygon-RNN},
  author = {Lluis Castrejon and Kaustav Kundu and Raquel Urtasun and Sanja Fidler},
  year   = {2017},
  eprint = {1704.05548},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/1704.05548}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/1704.05548.svg)](https://codeofpaper.com/paper/1704.05548)

Discussion