arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.09469cs.CVcs.LG

用于单目深度估计的虚拟增强NYU Depth V2数据集:我们需要人工增强吗?

Virtually Enriched NYU Depth V2 Dataset for Monocular Depth Estimation: Do We Need Artificial Augmentation?

ETH Zürich(苏黎世联邦理工学院);University of Würzburg(维尔茨堡大学);CAIDAS(卡伊达斯研究院);IFI(信息学研究所)

Dmitry Ignatov, Andrey Ignatov, Radu Timofte

AI总结:

本研究提出通过随机插入虚拟3D对象构建的虚拟增强数据集ANYU,验证了随机虚拟增强可提升不同架构深度神经网络的单目深度估计性能与泛化能力,并公开了两种配置的数据集。

摘要:

我们提出了ANYU,这是一款新型虚拟增强版NYU depth v2数据集,专为单目深度估计任务设计。与利用虚拟世界完整3D场景生成人工数据集的主流方法不同,ANYU是通过将虚拟现实(virtual reality,VR)对象的RGB-D表示融合到原始NYU depth v2图像中构建而成。我们特意未将每个生成的虚拟对象与真实图像内的适配纹理及合适位置进行匹配,而是随机分配纹理、位置、光照及其他渲染参数,以最大化训练数据的多样性,并证明正是这种随机性能够提升数据集的泛化能力。我们使用这款经过虚拟修改的数据集开展了大量实验,并在原始NYU depth v2和iBims-1基准上进行验证,结果表明ANYU能够提升架构差异显著的深度神经网络的单目深度估计性能与泛化能力,对当前最先进的VPD模型效果尤为突出。据我们所知,这是首项为单目深度估计任务,采用随机生成的虚拟3D对象增强真实世界数据集的研究。我们已将ANYU数据集公开,提供两种训练配置,分别包含10%和100%的额外合成增强RGB-D训练图像对,用于高效训练与虚拟增强的实证探索,获取地址为https://github.com/ABrain-One/ANYU。

英文摘要:

We present ANYU, a new virtually augmented version of the NYU depth v2 dataset, designed for monocular depth estimation. In contrast to the well-known approach where full 3D scenes of a virtual world are utilized to generate artificial datasets, ANYU was created by incorporating RGB-D representations of virtual reality objects into the original NYU depth v2 images. We specifically did not match each generated virtual object with an appropriate texture and a suitable location within the real-world image. Instead, an assignment of texture, location, lighting, and other rendering parameters was randomized to maximize a diversity of the training data, and to show that it is randomness that can improve the generalizing ability of a dataset. By conducting extensive experiments with our virtually modified dataset and validating on the original NYU depth v2 and iBims-1 benchmarks, we show that ANYU improves the monocular depth estimation performance and generalization of deep neural networks with considerably different architectures, especially for the current state-of-the-art VPD model. To the best of our knowledge, this is the first work that augments a real-world dataset with randomly generated virtual 3D objects for monocular depth estimation. We make our ANYU dataset publicly available in two training configurations with 10% and 100% additional synthetically enriched RGB-D pairs of training images, respectively, for efficient training and empirical exploration of virtual augmentation at https://github.com/ABrain-One/ANYU

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

↑