Acta Optica Sinica, Volume. 39, Issue 4, 0428004(2019)

Fully Convolutional Network Method of Semantic Segmentation of Class Imbalance Remote Sensing Images

Zhihuan Wu1,2、*, Yongming Gao3, Lei Li4, and Junshi Xue1
Author Affiliations
  • 1 Graduate School, Space Engineering University, Beijing 101416, China
  • 2 63883 Troops, Luoyang, Henan 471000, China
  • 3 School of Space Information, Space Engineering University, Beijing 101416, China
  • 4 Department of Electronic and Optical Engineering, Space Engineering University, Beijing 101416, China
  • show less

    A fully convolutional network (FCN) model based on U-Net is proposed to implement the semantic segmentation of remote sensing images with high resolution, in which the data standardization and data augmentation are adopted for data preprocessing. In addition, the Adam optimizer is used for the model training and the average Jaccard index is used as the evaluation metric. A weighted cross entropy loss function and an adaptive threshold algorithm are employed to improve the classification accuracy of small classes. The experimental results on the DSTL dataset show that the proposed method can increase the average Jaccard index of prediction results from 0.611 to 0.636, and produces an accurate end-to-end classification for high-resolution remote sensing images.

    Tools

    Get Citation

    Copy Citation Text

    Zhihuan Wu, Yongming Gao, Lei Li, Junshi Xue. Fully Convolutional Network Method of Semantic Segmentation of Class Imbalance Remote Sensing Images[J]. Acta Optica Sinica, 2019, 39(4): 0428004

    Download Citation

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

    Category: Remote Sensing and Sensors

    Received: Oct. 22, 2018

    Accepted: Dec. 29, 2018

    Published Online: May. 10, 2019

    The Author Email:

    DOI:10.3788/AOS201939.0428004

    Topics