Journal of South China University of Technology (Natural Science Edition) ›› 2020, Vol. 48 ›› Issue (12): 114-124.doi: 10.12141/j.issn.1000-565X.200365

• Artificial Intelligence Special • Previous Articles     Next Articles

W-Net-Based Segmentation for Remote Sensing Satellite Image of High Resolution

FAN Zizhu1 WANG Song1 ZHANG Hong1 SHI Linrui1 FU Jinwu1 LI Zhengming2   

  1. 1. School of Science,East China Jiaotong University,Nanchang 330013,Jiangxi,China; 2. Industrial Training Center,Guangdong Polytechnic Normal University,Guangzhou 510665,Guangdong,China
  • Received:2020-06-28 Revised:2020-08-09 Online:2020-12-25 Published:2020-12-01
  • Contact: 范自柱 ( 1975-) ,男,博士,教授,主要从事模式识别与数字图像处理研究。 E-mail:zzfan3@163.com
  • About author:范自柱 ( 1975-) ,男,博士,教授,主要从事模式识别与数字图像处理研究。
  • Supported by:

    Supported by the National Natural Science Foundation of China ( 61991401,61673097,61702117) and the Natural Science Foundation of Jiangxi Province ( 20192ACBL20010)

Abstract:

Remote sensing satellite image,which can provide accurate location information because of instantaneity and objectivity,has been widely used in agricultural production,environmental protection and other fields. In order to solve the difficulties in recognizing the mass remote sensing satellite images,the image segmentation method based on convolutional neural networks was used to extract the typical land spectral information and spatial information in the remote sensing image and to identify the remote sensing satellite image. Firstly,the experimental data was generated by clipping remote sensing ima-ge data and annotating data. The categories in the data were counted and oversampling was used to deal with the problem of data imbalance. Then,a new image segmentation network named W-NET was proposed to train the data by adding a top-down feature pyramid structure to the U-Net framework combining with global context module. Finally,large-scale remote sensing satellite images were recognized by image overlap strategy. Compared with three popular semantic segmentation networks and two special networks for remote sensing satellite image segmentation,our method achieves 74. 7% mean IoU and 95. 1% accuracy in 2017 AI classification and recognition contest of CCF satellite images,and achieves 69. 6% precision and 79. 9% recall in Massachusetts building segmentation task. W-Net has the highest accuracy and mean IoU among the six networks. Experimental results show that the feature pyramid structure and global context module can improve the segmentation accuracy of the semantic segmentation network,and this method is feasible for remote sensing satellite image segmentation.

Key words: remote sensing image, convolutional neural networks, image segmentation

CLC Number: