A combination of convolutional and graph neural networks for regularized road surface extraction

Jingjing Yan, Shunping Ji*, Yao Wei

*Corresponding author for this work

Research output: Contribution to journalArticleAcademicpeer-review

25 Citations (Scopus)
133 Downloads (Pure)

Abstract

Road surface extraction from high-resolution remote sensing images has many engineering applications; however, extracting regularized and smooth road surface maps that reach the human delineation level is a very challenging task, and substantial and time-consuming manual work is usually unavoidable. In this article, to solve this problem, we propose a novel regularized road surface extraction framework by introducing a graph neural network (GNN) for processing the road graph that is preconstructed from the easily accessible road centerlines. The proposed framework formulates the road surface extraction problem as two-sided width inference of the road graph and consists of a convolutional neural network (CNN)-based feature extractor and a GNN model for vertex attribute adjustment. The CNN extracts the high-level abstract features of each vertex in the graph as the input of the GNN and also the road boundary features that allow us to distinguish roads from the background. The GNN propagates and aggregates the features of the vertices in the graph to achieve global optimization of the regression of the regularized widths of the vertices. At the same time, a biased centerline map can also be corrected based on the width prediction result. To the best of the authors’ knowledge, this is the first study to have introduced a GNN to regularized human-level road surface extraction. The proposed method was evaluated on four diverse datasets, and the results show that the proposed method comprehensively outperforms the recent CNN-based segmentation methods and other regularization methods in the intersection over union (IoU) and smoothness score, and a visual check shows that a majority of the prediction results of the proposed method approach the human delineation level.
Original languageEnglish
Article number4409113
Number of pages13
JournalIEEE transactions on geoscience and remote sensing
Volume60
DOIs
Publication statusPublished - 15 Feb 2022

Keywords

  • Roads
  • Feature extraction
  • Remote sensing
  • Data mining
  • Task analysis
  • Convolutional neural networks
  • Satellites
  • ITC-ISI-JOURNAL-ARTICLE
  • ITC-HYBRID

Fingerprint

Dive into the research topics of 'A combination of convolutional and graph neural networks for regularized road surface extraction'. Together they form a unique fingerprint.

Cite this