CycleGAN Coastline Automatic Extraction Method Based on Dual Attention Mechanism

被引:0
|
作者
Lu Peng [1 ]
Zhang Na [1 ]
Zou Guoliang [1 ]
Wang Zhenhua [1 ]
Zheng Zongsheng [1 ]
机构
[1] Shanghai Ocean Univ, Coll Informat, Shanghai 201306, Peoples R China
关键词
image processing; remote sensing; cycle generative adversarial network; attention mechanism; cycle consistency loss; small sample;
D O I
10.3788/LOP202259.1210005
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
The pixel-level sea-land segmentation of remote sensing images is a basic work for coastline extraction. Owing to the dynamic changes in the coastline, obtaining accurate coastline marker datasets is difficult. In this study, Google Aerial Photo-Maps-paired samples were used to construct a paired dataset after the sea-land binarization processing of Google Maps. Thus, we proposed the dual attention mechanism-cycle generative adversarial network (CycleGAN) based on the CycleGAN model to solve the problem of fewer samples in the new dataset. The new model fully considers the structural similarity between remote sensing images and sealand binarized images, improves cycle consistency loss, and designs both channel and spatial attention modules to highlight salient features and regions to enhance the model's performance in small feature learning ability under sample training. Furthermore, we applied three evaluation indicators, i. e., mean square error, mean pixel accuracy, and mean intersection over union (MIoU), and compared our experimental results to those of the full convolutional neural network and DeepLab models under multiple-scale dataset training. Results show that the improved model conversion of the sea-land binarized images is more consistent with the true value images and the MIoU values are increased by at least 7% and 6%, respectively, verifying the effectiveness and feasibility of the proposed method.
引用
收藏
页数:11
相关论文
共 30 条
  • [1] Recycle-GAN: Unsupervised Video Retargeting
    Bansal, Aayush
    Ma, Shugao
    Ramanan, Deva
    Sheikh, Yaser
    [J]. COMPUTER VISION - ECCV 2018, PT V, 2018, 11209 : 122 - 138
  • [2] Chen L C, 2021, SEMANTIC IMAGE SEGME
  • [3] A New Framework for Automatic Airports Extraction from SAR Images Using Multi-Level Dual Attention Mechanism
    Chen, Lifu
    Tan, Siyu
    Pan, Zhouhao
    Xing, Jin
    Yuan, Zhihui
    Xing, Xuemin
    Zhang, Peng
    [J]. REMOTE SENSING, 2020, 12 (03)
  • [4] A New Deep Learning Algorithm for SAR Scene Classification Based on Spatial Statistical Modeling and Features Re-Calibration
    Chen, Lifu
    Cui, Xianliang
    Li, Zhenhong
    Yuan, Zhihui
    Xing, Jin
    Xing, Xuemin
    Jia, Zhiwei
    [J]. SENSORS, 2019, 19 (11)
  • [5] Identity-aware CycleGAN for face photo-sketch synthesis and recognition
    Fang, Yuke
    Deng, Weihong
    Du, Junping
    Hu, Jiani
    [J]. PATTERN RECOGNITION, 2020, 102
  • [6] Goodfellow IJ, 2014, ADV NEUR IN, V27, P2672
  • [7] Deep Convolutional Neural Networks for Hyperspectral Image Classification
    Hu, Wei
    Huang, Yangyu
    Wei, Li
    Zhang, Fan
    Li, Hengchao
    [J]. JOURNAL OF SENSORS, 2015, 2015
  • [8] [蒋科迪 Jiang Kedi], 2019, [测绘通报, Bulletin of Surveying and Mapping], P83
  • [9] DeepUNet: A Deep Fully Convolutional Network for Pixel-Level Sea-Land Segmentation
    Li, Ruirui
    Liu, Wenjie
    Yang, Lei
    Sun, Shihao
    Hu, Wei
    Zhang, Fan
    Li, Wei
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2018, 11 (11) : 3954 - 3962
  • [10] [李宗梅 Li Zongmei], 2017, [应用海洋学学报, Journal of Applied Oceanography], V36, P125