GA-Net: A geometry prior assisted neural network for road extraction

https://doi.org/10.1016/j.jag.2022.103004Get rights and content
Under a Creative Commons license
open access

Highlights

  • A novel Geometric Prior Assisted Neural Network (GA-Net) is proposed for automatic road extraction. The boundary information is introduced to the model, which strengthens the geometric details of road edges. The road junction information is jointly learned with the road surface information, which reduces the missing gap between road segments.

  • A feature fusion module is designed to bridge the semantic gap between multi-scale features maps. An attention module is designed to strengthen the road geometric information. The two light modules can be easily ensemble with other models.

  • We achieves better results than some SOTA algorithms for three popular datasets. And results of our model shows better completeness and connectivity.

Abstract

With geospatial intelligence research developing rapidly, automatic road extraction is becoming a fundamental and challenging task. Due to the special geometric structure and spectral information of road networks, existing methods suffer from incomplete and fractured results. In this work, a novel road extraction convolutional neural network, incorporating the road boundary details and road junction information via a dual-branch multi-task structure, is proposed to learn synergistic feature representations and strengthen road connectivity. Firstly, a BiFPN-based feature aggregation module is utilised to bridge the semantic gap between low-level and high-level feature maps, allowing multi-scale spatial details to be fully fused. Secondly, the boundary auxiliary branch, using a U-shaped network with a spatial-channel attention module, captures residential information for the backbone to enhance the subtleties of road edges. Thirdly, the node inferring branch models the road junction position jointly with the road surface, aiming to strengthen the topology structure and reduce the fragmented road segments. We perform experiments on three diverse road datasets, namely the DeepGlobe dataset, Massachusetts dataset, and SpaceNet dataset. The results demonstrate that our model shows an overall performance improvement over some SOTA algorithms and the IoU indicator achieves 3.86%, 0.79%, and 1.71% improvements over Unet on the three datasets, respectively.

Keywords

Road extraction
Remote sensing image
Deep learning
Multi-task learning

Data availability

Data will be made available on request.

Cited by (0)