Laser & Optoelectronics Progress, Volume. 58, Issue 20, 2010009(2021)

Semantic Segmentation of Synthetic Aperture Radar Images Based on U-Net and Capsule Network

Shaodi Jing1, Lingjuan Yu1、*, Yuehong Hu2, Zezhou Yang1, Zhongliang Lu1, and Xiaochun Xie3
Author Affiliations
  • 1School of Information Engineering, Jiangxi University of Science and Technology, Ganzhou, Jiangxi 341000, China
  • 2Guangzhou Wayful Technology Development Co., Ltd., Guangzhou, Guangdong 510200, China
  • 3School of Physics and Electronic Information, Gannan Normal University, Ganzhou, Jiangxi 341000, China
  • show less

    As a pixel-level classification technique, image semantic segmentation has been employed in the field of synthetic aperture radar (SAR) image interpretations. U-Net is an end-to-end image semantic segmentation network with a typical encoder-decoder architecture. Among them, the coding part mainly comprises a convolutional layer and a pooling layer, which can effectively extract the features of a target image; however, extracting information such as the target position and direction is difficult. Capsule network is a type of neural network that can obtain the target pose (position, size, and direction) and other information. Therefore, this study proposes an SAR image semantic segmentation method based on the U-Net and capsule network. Moreover, considering the small data set of SAR images, the U-Net encoder is designed to be identical to the visual geometry group (VGG16) to allow the trained VGG16 model to be directly transferred to the encoder. The effectiveness of the method is verified by conducting a segmentation experiment of building targets on two polarimetric SAR image data sets. Results show that the method can achieve improved precision, recall, F1-score, and intersection over union as well as reduce the training time of the network model when compared with the U-Net.

    Tools

    Get Citation

    Copy Citation Text

    Shaodi Jing, Lingjuan Yu, Yuehong Hu, Zezhou Yang, Zhongliang Lu, Xiaochun Xie. Semantic Segmentation of Synthetic Aperture Radar Images Based on U-Net and Capsule Network[J]. Laser & Optoelectronics Progress, 2021, 58(20): 2010009

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Nov. 20, 2020

    Accepted: Jan. 2, 2021

    Published Online: Oct. 13, 2021

    The Author Email: Yu Lingjuan (lingjuanyusmile@163.com)

    DOI:10.3788/LOP202158.2010009

    Topics