DOI number:10.1109/tgrs.2023.3335997

Journal:IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING

Key Words:Convolutional neural network,multitask learning,semantic segmentation,stereo image pairs,stereo matching

Abstract:Stereo matching and semantic segmentation are two significant tasks in remote sensing. Recently, deep learning approaches have been applied to these tasks separately. However, the lack of semantic supervision makes the training of stereo matching model susceptible to data disturbance, resulting in inferior generalization ability; foreground objects are sometimes confused with background pixels in RGB images, limiting the classification accuracy. By exploring the relationship between these two tasks, semantic stereo solves these problems simultaneously with multitask learning. Previous methods took semantic stereo as two parallel processing tasks, so they did not take full advantages of the additional information from both tasks and only obtained slight improvement. In this work, we designed a multitask learning framework semantic stereo network (S(2)Net). The proposed network generates cost volumes with feature maps supervised by semantic information to estimate disparity maps and fuses RGB-D feature maps to predict classification maps, therefore gathering multitask learning information. To enhance the performance of trained model, we also considered the continuity of disparity values and the duality of stereo image pairs in data augmentation. When applied in datasets without training, S(2)Net obtained 2.937% D1-Error in the WHU dataset, lower than 4.297% of the previous best method, depicting the generalization ability improvement from semantic supervision. In terms of semantic segmentation, the introduction of disparity maps increases the mean intersection over union (mIoU) from 61.375% to 69.096% in the US3D datasets. The experiments on the KITTI semantics benchmark show that our proposed method obtains 60.76% mIoU, achieving state-of-the-art among multitask learning methods.

Co-author:Tong Wang,Xianwei Li,Haobo Yang,Wenlin Zhou,Chanjuan He,Qing Wang

Indexed by:Journal paper

Correspondence Author:Xiaodong Zhang,Guanzhou Chen

Document Type:J

Volume:62

Page Number:1-13

ISSN No.:0196-2892

Translation or Not:no

Date of Publication:2023-11-28

Included Journals:SCI