Extraction of land covers from remote sensing images based on a deep learning model of NDVI-RSU-Net
Chen Yang,Jingwei Hou,Yanjuan Wang
DOI: https://doi.org/10.1007/s12517-021-08420-5
2021-09-26
Arabian Journal of Geosciences
Abstract:The acquisition of land-cover types is of great importance for land use and land planning. However, extracting accurate land-cover types using traditional methods such as visual interpretation, statistical analysis, and cluster analysis is difficult. Image segmentation method based on deep learning can effectively extract accurate land cover from a remote sensing image. First, a ReSidual U-block (RSU-Net) is first constructed in this paper according to the combination of U-Net, residual network, and attention mechanism. Then, a normalized vegetation index (NDVI) is added to the input feature channel to construct an NDVI-RSU-Net model to improve the extraction accuracy of land cover from the GaoFen-2 remote sensing images. Finally, the extraction results of the NDVI-RSU-Net model are compared with those of the full convolutional neural network (FCN), U-Net, LinkNet, VNet, RSU-Net, and ResU-Net models to verify the effectiveness of the NDVI-RSU-Net model in extracting land-cover types. The NDVI-RSU-Net model can extract the best classification accuracies of land-cover types with mean pixel accuracy (MPA) and mean intersection over union (MIOU) of 88.36% and 71.18%, respectively. The MPA and MIOU of the RSU-Net are higher than those of U-Net (85.34% and 65.10%, respectively), FCN (84.38% and 65.36%, respectively), ResU-Net (86.37% and 66.09%, respectively), LinkNet (81.89% and 55.43%, respectively), VNet (79.63% and 57.69%, respectively), and RSU-Net (87.24% and 67.69%, respectively). The combination of visible band, near infrared band, and NDVI is the best input channel of RSU-Net. The NDVI-RSU-Net model has a better segmentation effect on land-cover types from high-resolution remote sensing images than other deep learning algorithms.
geosciences, multidisciplinary