Multimedia >>

A Dual-Path and Lightweight Convolutional Neural Network for High-Resolution Aerial Image Segmentation

Title: A Dual-Path and Lightweight Convolutional Neural Network for High-Resolution Aerial Image Segmentation 

Author(s): Zhang, G (Zhang, Gang); Lei, T (Lei, Tao); Cui, Y (Cui, Yi); Jiang, P (Jiang, Ping) 

Source: ISPRS INTERNATIONAL JOURNAL OF GEO-INFORMATION  Volume: 8  Issue: 12  Article Number: 582  DOI: 10.3390/ijgi8120582  Published: DEC 2019   

Abstract: Semantic segmentation on high-resolution aerial images plays a significant role in many remote sensing applications. Although the Deep Convolutional Neural Network (DCNN) has shown great performance in this task, it still faces the following two challenges: intra-class heterogeneity and inter-class homogeneity. To overcome these two problems, a novel dual-path DCNN, which contains a spatial path and an edge path, is proposed for high-resolution aerial image segmentation. The spatial path, which combines the multi-level and global context features to encode the local and global information, is used to address the intra-class heterogeneity challenge. For inter-class homogeneity problem, a Holistically-nested Edge Detection (HED)-like edge path is employed to detect the semantic boundaries for the guidance of feature learning. Furthermore, we improve the computational efficiency of the network by employing the backbone of MobileNetV2. We enhance the performance of MobileNetV2 with two modifications: (1) replacing the standard convolution in the last four Bottleneck Residual Blocks (BRBs) with atrous convolution; and (2) removing the convolution stride of 2 in the first layer of BRBs 4 and 6. Experimental results on the ISPRS Vaihingen and Potsdam 2D labeling dataset show that the proposed DCNN achieved real-time inference speed on a single GPU card with better performance, compared with the state-of-the-art baselines. 

eISSN: 2220-9964 

  Copyright © The Institute of Optics And Electronics, The chinese Academy of Sciences
Address: Box 350, Shuangliu, Chengdu, Sichuan, China
Email:dangban@ioe.ac.cn Post Code: 610 209 备案号:蜀ICP备05022581号