arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.08368cs.CV

METER:一种用于单目深度估计的移动视觉Transformer架构

METER: a mobile vision transformer architecture for monocular depth estimation

Sapienza University of Rome(罗马第一大学)

L. Papa, P. Russo, I. Amerini

AI总结:

本文提出轻量级视觉Transformer架构METER,用于嵌入式设备上的单目深度估计,通过三种配置、新型损失函数和数据增强策略,在NYU Depth v2和KITTI数据集上超越现有轻量级方法,实现低延迟推理。

摘要:

深度估计是自主系统评估自身状态并感知周围环境所需的基础知识。近年来,深度学习的深度估计算法因其在克服主动深度传感系统局限性方面的潜在优势而引起了广泛关注。此外,由于单目相机成本低、体积小,研究人员将注意力集中在单目深度估计(MDE)上,即从单个RGB视频帧估计密集深度图。最先进的MDE模型通常依赖于高度深层且复杂的视觉Transformer(ViT)架构,这使得它们不适合在硬件受限的设备上进行快速推理。为此,在本文中,我们解决了在嵌入式设备上利用ViT进行MDE的问题。这些系统通常具有有限的内存容量和低功耗CPU/GPU的特点。我们提出了METER,一种新颖的轻量级视觉Transformer架构,能够在所考虑的嵌入式硬件(NVIDIA Jetson TX1和NVIDIA Jetson Nano)上实现最先进的估计和低延迟推理性能。我们提供了一种解决方案,包括METER的三种可选配置、一种用于平衡像素估计和图像细节重建的新型损失函数,以及一种新的数据增强策略,以改善最终的总体预测。所提出的方法在两个基准数据集(室内NYU Depth v2和室外KITTI)上优于先前的轻量级工作。

英文摘要:

Depth estimation is a fundamental knowledge for autonomous systems that need to assess their own state and perceive the surrounding environment. Deep learning algorithms for depth estimation have gained significant interest in recent years, owing to the potential benefits of this methodology in overcoming the limitations of active depth sensing systems. Moreover, due to the low cost and size of monocular cameras, researchers have focused their attention on monocular depth estimation (MDE), which consists in estimating a dense depth map from a single RGB video frame. State of the art MDE models typically rely on vision transformers (ViT) architectures that are highly deep and complex, making them unsuitable for fast inference on devices with hardware constraints. Purposely, in this paper, we address the problem of exploiting ViT in MDE on embedded devices. Those systems are usually characterized by limited memory capabilities and low-power CPU/GPU. We propose METER, a novel lightweight vision transformer architecture capable of achieving state of the art estimations and low latency inference performances on the considered embedded hardwares: NVIDIA Jetson TX1 and NVIDIA Jetson Nano. We provide a solution consisting of three alternative configurations of METER, a novel loss function to balance pixel estimation and reconstruction of image details, and a new data augmentation strategy to improve the overall final predictions. The proposed method outperforms previous lightweight works over the two benchmark datasets: the indoor NYU Depth v2 and the outdoor KITTI.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑