The availability of large-scale annotated datasets has enabled Fully-Convolutional Neural Networks to reach outstanding performance on road extraction in aerial images. However, high-quality pixel-level annotation is expensive to produce and even manually labeled data often contains topological errors. Trading off quality for quantity, many datasets rely on already available yet noisy labels, for example from OpenStreetMap. In this paper, we explore the training of custom U-Nets built with ResNet and DenseNet backbones using noise-aware losses that are robust towards label omission and registration noise. We perform an extensive evaluation of standard and noise-aware losses, including a novel Bootstrapped DICE-Coefficient loss, on two challenging road segmentation benchmarks. Our losses yield a consistent improvement in overall extraction quality and exhibit a strong capacity to cope with severe label noise. Our method generalizes well to two other fine-grained topology delineation tasks: surface crack detection for quality inspection and cell membrane extraction in electron microscopy imagery.
|Publikationsstatus||Veröffentlicht - 12 Jan. 2021|
|Veranstaltung||25th International Conference on Pattern Recognition - Milano, Virtuell, Italien|
Dauer: 10 Jan. 2021 → 15 Feb. 2021
|Konferenz||25th International Conference on Pattern Recognition|
|Zeitraum||10/01/21 → 15/02/21|
Fields of Expertise
- Information, Communication & Computing