High-Resolution Remote Sensing Image Segmentation Framework Based on Attention Mechanism and Adaptive Weighting
Open Access
- 7 April 2021
- journal article
- research article
- Published by MDPI AG in ISPRS International Journal of Geo-Information
- Vol. 10 (4) , 241
Abstract
Semantic segmentation has been widely used in the basic task of extracting information from images. Despite this progress, there are still two challenges: (1) it is difficult for a single-size receptive field to acquire sufficiently strong representational features, and (2) the traditional encoder-decoder structure directly integrates the shallow features with the deep features. However, due to the small number of network layers that shallow features pass through, the feature representation ability is weak, and noise information will be introduced to affect the segmentation performance. In this paper, an Adaptive Multi-Scale Module (AMSM) and Adaptive Fuse Module (AFM) are proposed to solve these two problems. AMSM adopts the idea of channel and spatial attention and adaptively fuses three-channel branches by setting branching structures with different void rates, and flexibly generates weights according to the content of the image. AFM uses deep feature maps to filter shallow feature maps and obtains the weight of deep and shallow feature maps to filter noise information in shallow feature maps effectively. Based on these two symmetrical modules, we have carried out extensive experiments. On the ISPRS Vaihingen dataset, the F1-score and Overall Accuracy (OA) reached 86.79% and 88.35%, respectively.Keywords
This publication has 42 references indexed in Scilit:
- High-Resolution Aerial Image Labeling With Convolutional Neural NetworksIEEE Transactions on Geoscience and Remote Sensing, 2017
- Pyramid Scene Parsing Network2017
- RefineNet: Multi-path Refinement Networks for High-Resolution Semantic Segmentation2017
- DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFsIEEE Transactions on Pattern Analysis and Machine Intelligence, 2017
- Semantic Classification of Urban Trees Using Very High Resolution Satellite ImageryIEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 2017
- SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image SegmentationIEEE Transactions on Pattern Analysis and Machine Intelligence, 2017
- Dense Semantic Labeling of Subdecimeter Resolution Images With Convolutional Neural NetworksIEEE Transactions on Geoscience and Remote Sensing, 2016
- Semantic Image Segmentation via Deep Parsing Network2015
- U-Net: Convolutional Networks for Biomedical Image Segmentation2015
- Segment-Based Land Cover Mapping of a Suburban Area—Comparison of High-Resolution Remotely Sensed Datasets Using Classification Trees and Test Field PointsRemote Sensing, 2011