END TO END TRAINABLE ACTIVE CONTOURS VIA DIFFERENTIABLE RENDERING

Shir Gur, Tal Shaharabany, Lior Wolf

Research output: Contribution to conferencePaperpeer-review

11 Scopus citations

Abstract

We present an image segmentation method that iteratively evolves a polygon. At each iteration, the vertices of the polygon are displaced based on the local value of a 2D shift map that is inferred from the input image via an encoder-decoder architecture. The main training loss that is used is the difference between the polygon shape and the ground truth segmentation mask. The network employs a neural renderer to create the polygon from its vertices, making the process fully differentiable. We demonstrate that our method outperforms the state of the art segmentation networks and deep active contour solutions in a variety of benchmarks, including medical imaging and aerial images. Our code is available at https://github.com/shirgur/ACDRNet.

Original languageEnglish
StatePublished - 2020
Event8th International Conference on Learning Representations, ICLR 2020 - Addis Ababa, Ethiopia
Duration: 30 Apr 2020 → …

Conference

Conference8th International Conference on Learning Representations, ICLR 2020
Country/TerritoryEthiopia
CityAddis Ababa
Period30/04/20 → …

Funding

FundersFunder number
European Commission
Horizon 2020ERC CoG 725974

    Fingerprint

    Dive into the research topics of 'END TO END TRAINABLE ACTIVE CONTOURS VIA DIFFERENTIABLE RENDERING'. Together they form a unique fingerprint.

    Cite this