BiReNet: Bilateral Network with Feature Aggregation and Edge Detection for Remote Sensing Images Road Extraction
摘要
Extracting roads from remote sensing images is a significant and challenging topic. Most existing advanced methods perform well in general scenarios but cannot cope with complex scenarios, such as obscured and covered roads. This paper proposes a Bilateral Road Extraction Network (BiReNet) consisting of an edge detection branch and a road extraction branch. Firstly, we adopt an effective and efficient LinkNet architecture in the road extraction branch and add an extra Feature Fusion Module (FFM) behind the skip connection part. The FFM is capable of efficiently aggregating the features at the same level from the decoder and the corresponding encoder, thus preserving abundant spatial information and amplifying comprehension of road structures. Secondly, we design a novel Edge Detection Module (EDM) in the edge detection branch to enhance the road edge features by capturing the gradient information using Pixel Difference Convolution (PDC), which enables fine-grained constraints on the road extraction and improves the accuracy and connectivity of the road. Extensive experiments on two publicly available road datasets show that BiReNet performs favorably against other state-of-the-art remote sensing road extraction methods and demonstrates stronger robustness in complex scenarios. Specifically, for a 1024 \(\,\times \,\) 1024 input, BiReNet achieved 0.6769 IoU on the DeepGlobe road dataset and 0.6072 IoU on the Massachusetts road dataset with a speed of 24 FPS on one GeForce GTX3090. The code is available at https://github.com/LPeng625/BiReNet.