MCTN-Net: A Multiclass Transportation Network Extraction Method Combining Orientation and Semantic Features

Chenglin Shao,Huifang Li,Huanfeng Shen
DOI: https://doi.org/10.1109/lgrs.2024.3372194
IF: 5.343
2024-03-15
IEEE Geoscience and Remote Sensing Letters
Abstract:Transportation network extraction based on deep learning has become a hotspot. However, the existing models all aim to distinguish between background and transportation networks, while ignoring the class attributes within the transportation networks. In this letter, we propose a multiclass transportation network extraction network (MCTN-Net) to simultaneously extract railways, roadways, trails, and bridges. Inspired by multitask learning, the network first extracts the orientation and semantic information together by the use of a dense feature shared encoder (DFSE). The orientation and semantic features are then fused in the orientation-guided stacking module (OGSM) to enhance the connection between transportation network pixels. Furthermore, a semantic refinement branch (SRB) is designed to improve the ability to classify different transportation network types through deep supervised fusion and class attention. A multiclass transportation network dataset (MCTN dataset) was constructed and used in the experiments. The experiential results indicate that the proposed method achieves a mean intersection over union (MIoU) of 64.29% and a frequency-weighted intersection over union (FWIoU) of 71.20% without the background, which is significantly better than the other road extraction models and semantic segmentation methods. The code and dataset are available at https://github.com/fzzfRS/MCTN-Net.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?