DHRNet: A Dual-Branch Hybrid Reinforcement Network for Semantic Segmentation of Remote Sensing Images

被引:9
作者
Bai, Qinyan [1 ,2 ]
Luo, Xiaobo [1 ,2 ]
Wang, Yaxu [1 ,2 ]
Wei, Tengfei [1 ,2 ]
机构
[1] Chongqing Univ Posts & Telecommun, Sch Comp Sci & Technol, Chongqing 400065, Peoples R China
[2] Chongqing Engn Res Ctr Spatial Big Data Intelligen, Chongqing 400065, Peoples R China
关键词
Global context modeling; multiscale feature extraction; remote sensing; semantic segmentation; CONVOLUTIONAL NEURAL-NETWORKS; LAND-USE; TEXTURE; CLASSIFICATION; EXTRACTION; COLOR;
D O I
10.1109/JSTARS.2024.3357216
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
In the field of remote sensing image processing, semantic segmentation has always been a hot research topic. Currently, deep convolutional neural networks (DCNNs) are the mainstream methods for the semantic segmentation of remote sensing image (RSI). There are two commonly used semantic segmentation methods based on DCNNs: multiscale feature extraction based on deep-level features, and global modeling. The former can better extract object features of different scales in complex scenes. However, this method lacks sufficient spatial information, resulting in poor edge segmentation ability. The latter can effectively solve the problem of limited receptive field in DCNNs obtaining more comprehensive feature extraction results. Unfortunately, this method is prone to misclassification, resulting in incorrect predictions of local pixels. To address these issues, we propose the dual-branch hybrid reinforcement network (DHRNet) for more precise semantic segmentation of RSI. This model is a dual-branch parallel structure with a multiscale feature extraction branch and a global context and detail enhancement branch. This structure decomposes the complex semantic segmentation task, allowing each branch to extract features with different emphases while retaining sufficient spatial information. The results of both branches are fused to obtain a more comprehensive segmentation result. After conducting extensive experiments on three publicly available RSI datasets, ISPRS Potsdam, ISPRS Vaihingen, and LoveDA, DHRNet demonstrates excellent results with the mean intersection over union of 86.97%, 83.53%, and 54.48% on the three datasets, respectively.
引用
收藏
页码:4176 / 4193
页数:18
相关论文
共 69 条
[21]   A fast algorithm to delineate watershed boundaries for simple geometries [J].
Haag, Scott ;
Schwartz, Daniel ;
Shakibajahromi, Bahareh ;
Campagna, Michael ;
Shokoufandeh, Ali .
ENVIRONMENTAL MODELLING & SOFTWARE, 2020, 134
[22]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[23]   Grayscale image segmentation using color space [J].
Horiuchi, T .
IEICE TRANSACTIONS ON INFORMATION AND SYSTEMS, 2006, E89D (03) :1231-1237
[24]  
Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/TPAMI.2019.2913372, 10.1109/CVPR.2018.00745]
[25]   MPViT : Multi-Path Vision Transformer for Dense Prediction [J].
Lee, Youngwan ;
Kim, Jonghee ;
Willette, Jeffrey ;
Hwang, Sung Ju .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, :7277-7286
[26]   Spectral-Spatial Hyperspectral Image Segmentation Using Subspace Multinomial Logistic Regression and Markov Random Fields [J].
Li, Jun ;
Bioucas-Dias, Jose M. ;
Plaza, Antonio .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2012, 50 (03) :809-823
[27]   Multiattention Network for Semantic Segmentation of Fine-Resolution Remote Sensing Images [J].
Li, Rui ;
Zheng, Shunyi ;
Zhang, Ce ;
Duan, Chenxi ;
Su, Jianlin ;
Wang, Libo ;
Atkinson, Peter M. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[28]   Deep Learning for Hyperspectral Image Classification: An Overview [J].
Li, Shutao ;
Song, Weiwei ;
Fang, Leyuan ;
Chen, Yushi ;
Ghamisi, Pedram ;
Benediktsson, Jon Atli .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2019, 57 (09) :6690-6709
[29]   A Survey of Convolutional Neural Networks: Analysis, Applications, and Prospects [J].
Li, Zewen ;
Liu, Fan ;
Yang, Wenjie ;
Peng, Shouheng ;
Zhou, Jun .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2022, 33 (12) :6999-7019
[30]   Feature Pyramid Networks for Object Detection [J].
Lin, Tsung-Yi ;
Dollar, Piotr ;
Girshick, Ross ;
He, Kaiming ;
Hariharan, Bharath ;
Belongie, Serge .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :936-944