Unsupervised Domain Adaptation Depth Estimation Based on Self-attention Mechanism and Edge Consistency Constraints

被引:0
|
作者
Guo, Peng [1 ]
Pan, Shuguo [1 ]
Hu, Peng [1 ]
Pei, Ling [2 ]
Yu, Baoguo [3 ]
机构
[1] Southeast Univ, Sch Instrument Sci & Engn, Nanjing 210096, Peoples R China
[2] Shanghai Jiao Tong Univ, Sch Elect Informat & Elect Engn, Shanghai, Peoples R China
[3] State Key Lab Satellite Nav Syst & Equipment Techn, Shijiazhuang, Peoples R China
关键词
Depth estimation; Unsupervised domain adaptation; Self-attention mechanism; Edge consistency constraints; NETWORK;
D O I
10.1007/s11063-024-11621-0
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In the unsupervised domain adaptation (UDA) (Akada et al. Self-supervised learning of domain invariant features for depth estimation, in: 2022 IEEE/CVF winter conference on applications of computer vision (WACV), pp 3377-3387 (2022). 10.1109/WACV51458.2022.00107) depth estimation task, a new adaptive approach is to use the bidirectional transformation network to transfer the style between the target and source domain inputs, and then train the depth estimation network in their respective domains. However, the domain adaptation process and the style transfer may result in defects and biases, often leading to depth holes and instance edge depth missing in the target domain's depth output. To address these issues, We propose a training network that has been improved in terms of model structure and supervision constraints. First, we introduce a edge-guided self-attention mechanism in the task network of each domain to enhance the network's attention to high-frequency edge features, maintain clear boundaries and fill in missing areas of depth. Furthermore, we utilize an edge detection algorithm to extract edge features from the input of the target domain. Then we establish edge consistency constraints between inter-domain entities in order to narrow the gap between domains and make domain-to-domain transfers easier. Our experimental demonstrate that our proposed method effectively solve the aforementioned problem, resulting in a higher quality depth map and outperforming existing state-of-the-art methods.
引用
收藏
页数:19
相关论文
共 50 条
  • [1] Unsupervised Monocular Depth Estimation and Visual Odometry Based on Generative Adversarial Network and Self-attention Mechanism
    Ye, Xingyu
    He, Yuanlie
    Ru, Shaonan
    [J]. Jiqiren/Robot, 2021, 43 (02): : 203 - 213
  • [2] Unsupervised Pansharpening Based on Self-Attention Mechanism
    Qu, Ying
    Baghbaderani, Razieh Kaviani
    Qi, Hairong
    Kwan, Chiman
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (04): : 3192 - 3208
  • [3] Indoor Depth Completion with Boundary Consistency and Self-Attention
    Huang, Yu-Kai
    Wu, Tsung-Han
    Liu, Yueh-Cheng
    Hsu, Winston H.
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW), 2019, : 1070 - 1078
  • [4] UNSUPERVISED DOMAIN ADAPTION FOR REMOTE SENSING SEMANTIC SEGMENTATION WITH SELF-ATTENTION MECHANISM
    Liu, Keming
    Liu, Fang
    Liu, Jia
    Xiao, Liang
    Tang, Xu
    [J]. IGARSS 2023 - 2023 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM, 2023, : 6916 - 6919
  • [5] Unsupervised domain adaptation with self-attention for post-disaster building damage detection
    Li, Yundong
    Lin, Chen
    Li, Hongguang
    Hu, Wei
    Dong, Han
    Liu, Yi
    [J]. NEUROCOMPUTING, 2020, 415 : 27 - 39
  • [6] SADD: Generative Adversarial Networks via Self-attention and Dual Discriminator in Unsupervised Domain Adaptation
    Dai, Zaiyan
    Yang, Jun
    Fan, Anfei
    Jia, Jinyin
    Chen, Junfan
    [J]. PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2023, PT VIII, 2024, 14432 : 473 - 484
  • [7] Unsupervised image-to-image translation by semantics consistency and self-attention
    Zhang Zhibin
    Xue Wanli
    Fu Guokai
    [J]. OPTOELECTRONICS LETTERS, 2022, 18 (03) : 175 - 180
  • [8] Unsupervised image-to-image translation by semantics consistency and self-attention
    Zhibin Zhang
    Wanli Xue
    Guokai Fu
    [J]. Optoelectronics Letters, 2022, 18 : 175 - 180
  • [9] Unsupervised image-to-image translation by semantics consistency and self-attention
    ZHANG Zhibin
    XUE Wanli
    FU Guokai
    [J]. Optoelectronics Letters, 2022, 18 (03) : 175 - 180
  • [10] Correlation alignment with attention mechanism for unsupervised domain adaptation
    Chen, Rong
    Ren, Chongguang
    [J]. WEB INTELLIGENCE, 2020, 18 (04) : 261 - 267