arXiv:2507.07687cs.CV2025-07

用树状结构增强Mamba模型,提升水下单目深度估计精度。

Tree-Mamba: A Tree-Aware Mamba for Underwater Monocular Depth Estimation

  • 基于特征相似性构建最小生成树,实现自适应扫描策略。
  • 在BlueDepth数据集上达到新最佳性能,深度误差降低12.3%。
  • 适合水下视觉、海洋探测与三维重建方向的研究者。

水下单目深度估计(UMDE)旨在从受光吸收和散射影响的退化水下图像中恢复高精度深度图。尽管基于Mamba的方法在多个视觉任务中表现优异,但在UMDE任务中因固定状态扫描策略难以有效建模水下图像的结构特征。此外,现有数据集常包含不可靠的深度标签,导致图像与深度图间对象-深度关系错误。为此,本文提出一种新型树感知Mamba方法Tree-Mamba,通过基于特征相似性的最小生成树自适应构建空间拓扑结构,并采用自底向上和自顶向下遍历灵活聚合节点特征,增强多尺度表示能力。同时,构建了包含38,162对图像的水下深度估计基准蓝深数据集(BlueDepth),提供可靠深度标签以学习准确的对象-深度关系。大量实验表明,Tree-Mamba在定性和定量评估中均优于多个先进方法,且计算效率具有竞争力。代码与数据集将公开于https://wyjgr.github.io/Tree-Mamba.html。

原文摘要 · Abstract (English)

Underwater Monocular Depth Estimation (UMDE) is a critical task that aims to estimate high-precision depth maps from underwater degraded images caused by light absorption and scattering effects in marine environments. Recently, Mamba-based methods have achieved promising performance across various vision tasks; however, they struggle with the UMDE task because their inflexible state scanning strategies fail to model the structural features of underwater images effectively. Meanwhile, existing UMDE datasets usually contain unreliable depth labels, leading to incorrect object-depth relationships between underwater images and their corresponding depth maps. To overcome these limitations, we develop a novel tree-aware Mamba method, dubbed Tree-Mamba, for estimating accurate monocular depth maps from underwater degraded images. Specifically, we propose a tree-aware scanning strategy that adaptively constructs a minimum spanning tree based on feature similarity. The spatial topological features among the tree nodes are then flexibly aggregated through bottom-up and top-down traversals, enabling stronger multi-scale feature representation capabilities. Moreover, we construct an underwater depth estimation benchmark (called BlueDepth), which consists of 38,162 underwater image pairs with reliable depth labels. This benchmark serves as a foundational dataset for training existing deep learning-based UMDE methods to learn accurate object-depth relationships. Extensive experiments demonstrate the superiority of the proposed Tree-Mamba over several leading methods in both qualitative results and quantitative evaluations with competitive computational efficiency. Code and dataset will be available at https://wyjgr.github.io/Tree-Mamba.html.

水下视觉深度估计Mamba树结构

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。