Promoting Connectivity of Network-Like Structures by Enforcing Region Separation

1Computer Vision Laboratory, EPFL 2Stanford University
EPFL logo Stanford logo
TOPOLoss

Our new TopoLoss enforces the network to reconstruct network-like structures with much better connectivity.

Abstract

We propose a novel, connectivity-oriented loss function for training deep convolutional networks to reconstruct network-like structures, like roads and irrigation canals, from aerial images. The main idea behind our loss is to express the connectivity of roads, or canals, in terms of disconnections that they create between background regions of the image. In simple terms, a gap in the predicted road causes two background regions, that lie on the opposite sides of a ground truth road, to touch in prediction. Our loss function is designed to prevent such unwanted connections between background regions, and therefore close the gaps in predicted roads. It also prevents predicting false positive roads and canals by penalizing unwarranted disconnections of background regions. In order to capture even short, dead-ending road segments, we evaluate the loss in small image crops. We show, in experiments on two standard road benchmarks and a new data set of irrigation canals, that convnets trained with our loss function recover road connectivity so well, that it suffices to skeletonize their output to produce state of the art maps. A distinct advantage of our approach is that the loss can be plugged in to any existing training setup without further modifications.

BibTeX

@inproceedings{Oner21,
  author = {D. Oner and M. Kozi{\'n}ski and L. Citraro and N. C. Dadap and A. G. Konings and P. Fua},
  title = {Promoting Connectivity of Network-Like Structures by Enforcing Region Separation},
  booktitle = PAMI,
  year = 2021
}
-->