arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.02494cs.CV

Plane2Depth:面向单目深度估计的分层自适应平面引导方法

Plane2Depth: Hierarchical Adaptive Plane Guidance for Monocular Depth Estimation

Deep Space Exploration Laboratory(深空探测实验室);School of Information Science and Technology, University of Science and Technology of China(中国科学技术大学信息科学技术学院);University of Science and Technology of China(中国科学技术大学)

Li Liu, Ruijie Zhu, Jiacheng Deng, Ziyang Song, Wenfei Yang, Tianzhu Zhang

AI总结:

针对单目深度估计在室内低纹理区域表现不佳的问题,提出Plane2Depth分层框架,通过平面引导生成器和自适应查询聚合模块利用平面信息,在多个数据集上取得优异性能且泛化性良好。

摘要:

单目深度估计旨在从单张图像推断稠密深度图,是计算机视觉中一项基础且普遍的任务。以往许多工作通过精心设计的网络结构取得了出色的深度估计结果,但它们通常忽略平面信息,因此在室内场景的低纹理区域表现不佳。本文提出Plane2Depth,在分层框架内自适应利用平面信息改进深度预测。具体而言,在提出的平面引导深度生成器(PGDG)中,我们设计了一组平面查询作为原型,对场景中的平面进行软建模并预测逐像素平面系数,随后可通过针孔相机模型将预测的平面系数转换为公制深度值。在提出的自适应平面查询聚合(APGA)模块中,我们引入一种新颖的特征交互方法,以自顶向下的方式改进多尺度平面特征的聚合。大量实验表明,我们的方法能够取得优异性能,尤其在低纹理或重复纹理区域表现突出。此外,在相同骨干网络下,我们的方法在NYU-Depth-v2数据集上优于现有最优方法,在KITTI数据集上取得了与现有最优方法相当的结果,并且能够有效泛化到未见过的场景。

英文摘要:

Monocular depth estimation aims to infer a dense depth map from a single image, which is a fundamental and prevalent task in computer vision. Many previous works have shown impressive depth estimation results through carefully designed network structures, but they usually ignore the planar information and therefore perform poorly in low-texture areas of indoor scenes. In this paper, we propose Plane2Depth, which adaptively utilizes plane information to improve depth prediction within a hierarchical framework. Specifically, in the proposed plane guided depth generator (PGDG), we design a set of plane queries as prototypes to softly model planes in the scene and predict per-pixel plane coefficients. Then the predicted plane coefficients can be converted into metric depth values with the pinhole camera model. In the proposed adaptive plane query aggregation (APGA) module, we introduce a novel feature interaction approach to improve the aggregation of multi-scale plane features in a top-down manner. Extensive experiments show that our method can achieve outstanding performance, especially in low-texture or repetitive areas. Furthermore, under the same backbone network, our method outperforms the state-of-the-art methods on the NYU-Depth-v2 dataset, achieves competitive results with state-of-the-art methods KITTI dataset and can be generalized to unseen scenes effectively.

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

↑