欢迎访问《图学学报》

图学学报 ›› 2026, Vol. 47 ›› Issue (3): 553-563.DOI: 10.11996/JG.j.2095-302X.2026030553

• 图像处理与计算机视觉 • 上一篇    下一篇

融合层次化双流注意力的单目深度估计方法

吴文欢1,2,3(), 王文舒1, 王舒鳌1   

  1. 1 湖北汽车工业学院智能网联汽车学院湖北 十堰 442002
    2 湖北汽车工业学院汽车动力传动与电子控制湖北省重点实验室湖北 十堰 442002
    3 湖北汽车工业学院空地群智协同技术与应用十堰市重点实验室湖北 十堰 442002
  • 收稿日期:2025-07-04 接受日期:2026-02-02 出版日期:2026-06-30 发布日期:2026-06-30
  • 通讯作者:吴文欢,E-mail:wuwenhuan5@163.com
  • 基金资助:
    湖北省自然科学基金创新发展联合基金项目(2025AFD239);汽车动力传动与电子控制湖北省重点实验室开放基金项目(ZDK12025A02)

Monocular depth estimation method with hierarchical dual-stream attention

WU Wenhuan1,2,3(), WANG Wenshu1, WANG Shuao1   

  1. 1 School of Intelligent Connected Vehicle, Hubei University of Automotive Technology, Shiyan Hubei 442002, China
    2 Key Laboratory of Automotive Power Train and Electronics, Hubei University of Automotive Technology, Shiyan Hubei 442002, China
    3 Shiyan Key Laboratory of Air-Ground Crowd Cooperation Technology and Application, Hubei University of Automotive Technology, Shiyan Hubei 442002, China
  • Received:2025-07-04 Accepted:2026-02-02 Published:2026-06-30 Online:2026-06-30
  • Contact: WU Wenhuan,E-mail:wuwenhuan5@163.com
  • Supported by:
    Innovation and Development Joint Fund Project of Natural Science Foundation of Hubei Province(2025AFD239);Open Fund Project of Key Laboratory of Automotive Power Train and Electronics (Hubei University of Automotive Technology)(ZDK12025A02)

摘要:

单目深度估计因其在自动驾驶、三维重建等领域具有广泛地应用前景而持续受到关注。然而,现有方法在多尺度特征融合与深度建模方面仍存在不足,难以同时兼顾局部几何细节刻画与全局结构一致性,对不同场景下深度尺度分布的自适应能力有限。针对上述问题,构建了一种融合层次化双流注意力与自适应深度离散建模的单目深度估计框架。编码阶段采用 Swin Transformer 构建金字塔式多尺度特征表示,以增强对局部与全局信息的联合建模能力;解码阶段设计层次化双流注意力融合网络,在逐级重建过程中并行建模局部细节感知与全局上下文语义,并通过动态权重调制与交叉注意力机制实现特征的自适应融合。同时,引入深度恢复模块,将深度估计建模为分类与回归相结合的任务,通过预测离散深度分布并自适应学习分箱中心,以概率加权方式生成连续深度结果,在保证深度预测连续性的同时,有效保持场景中深度关系的一致性。实验结果表明,该方法在 KITTI 数据集上的AbsRel 为 0.048、SqRel 为 0.147、log10 为 0.020、δ? 为 0.980,并在 NYU Depth V2 数据集上展现出良好的泛化能力,验证了该方法在复杂场景和多尺度深度分布条件下的有效性与鲁棒性。

关键词: 深度学习, 单目深度估计, 金字塔结构, 注意力机制, 特征融合

Abstract:

Monocular depth estimation has attracted sustained attention due to its broad application prospects in autonomous driving, 3D reconstruction, and related fields. However, existing methods still exhibit deficiencies in multi-scale feature fusion and depth modeling, which makes it difficult to simultaneously capture fine-grained local geometric details and maintain global structural consistency, and have limited adaptability to depth scale distributions across different scenes. To address these issues, a monocular depth estimation framework integrating hierarchical dual-stream attention and adaptive depth discretization was developed. In the encoding stage, a Swin Transformer was employed to construct pyramid multi-scale feature representations, enhancing the joint modeling of local and global information. In the decoding stage, a hierarchical dual-stream attention fusion network was designed to model local detail perception and global contextual semantics in parallel during progressive reconstruction, in which adaptive feature fusion was achieved through dynamic weight modulation and cross-attention mechanisms. Meanwhile, a depth recovery module was introduced to formulate depth estimation as a joint classification-regression task. By predicting a discrete depth distribution and adaptively learning the bin centers, continuous depth values were generated via probability-weighted aggregation, which ensured the continuity of depth prediction while effectively maintaining the consistency of depth relationships within the scene. Experimental results demonstrated that the proposed method achieved state-of-the-art performance on the KITTI dataset, with an AbsRel of 0.048, a SqRel of 0.147, a log10 error of 0.020, and a δ? accuracy of 0.980, and exhibited favorable generalization capability on the NYU Depth V2 dataset, validating its effectiveness and robustness under complex scenes and multi-scale depth distributions.

Key words: deep learning, monocular depth estimation, pyramid structure, attention mechanism, feature fusion

中图分类号: