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

PoseGen:学习使用NeRF生成3D人体姿态数据集

PoseGen: Learning to Generate 3D Human Pose Dataset with NeRF

University of British Columbia(不列颠哥伦比亚大学)

Mohsen Gholami, Rabab Ward, Z. Jane Wang

AI总结:

提出PoseGen框架,利用NeRF生成3D人体姿态数据集,通过反馈损失优化数据分布以提升预训练模型鲁棒性,实验显示在四个数据集上平均相对提升6%。

摘要:

本文提出了一种使用神经辐射场(NeRF)生成3D人体姿态数据集的端到端框架。公开数据集在人体姿态和相机视角方面的多样性通常有限,这主要是由于收集3D人体姿态数据需要大量资源。因此,在公开数据集上训练的姿态估计器在应用于未见过的分布外样本时性能显著下降。以往的工作通过生成2D-3D姿态对或渲染大量随机数据来增强公开数据集。这类方法要么忽略了图像渲染,要么为预训练模型生成次优的数据集。在这里,我们提出了PoseGen,它通过来自给定预训练姿态估计器的反馈损失来学习生成数据集(人体3D姿态和图像)。与现有技术相比,我们生成的数据旨在提高预训练模型的鲁棒性。PoseGen的目标是学习一种数据分布,该分布能够最大化给定预训练模型的预测误差。由于学习到的数据分布包含预训练模型的分布外(OOD)样本,从这样的分布中采样数据以进一步微调预训练模型,可以提高模型的泛化能力。这是首个提出使用NeRF进行3D人体数据生成的工作。NeRF是数据驱动的,不需要人体的3D扫描。因此,使用NeRF进行数据生成是为用户特定数据生成提供便利的新方向。我们的大量实验表明,所提出的PoseGen在四个数据集上平均相对改进了两个基线模型(SPIN和HybrIK)6%。

英文摘要:

This paper proposes an end-to-end framework for generating 3D human pose datasets using Neural Radiance Fields (NeRF). Public datasets generally have limited diversity in terms of human poses and camera viewpoints, largely due to the resource-intensive nature of collecting 3D human pose data. As a result, pose estimators trained on public datasets significantly underperform when applied to unseen out-of-distribution samples. Previous works proposed augmenting public datasets by generating 2D-3D pose pairs or rendering a large amount of random data. Such approaches either overlook image rendering or result in suboptimal datasets for pre-trained models. Here we propose PoseGen, which learns to generate a dataset (human 3D poses and images) with a feedback loss from a given pre-trained pose estimator. In contrast to prior art, our generated data is optimized to improve the robustness of the pre-trained model. The objective of PoseGen is to learn a distribution of data that maximizes the prediction error of a given pre-trained model. As the learned data distribution contains OOD samples of the pre-trained model, sampling data from such a distribution for further fine-tuning a pre-trained model improves the generalizability of the model. This is the first work that proposes NeRFs for 3D human data generation. NeRFs are data-driven and do not require 3D scans of humans. Therefore, using NeRF for data generation is a new direction for convenient user-specific data generation. Our extensive experiments show that the proposed PoseGen improves two baseline models (SPIN and HybrIK) on four datasets with an average 6% relative improvement.

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

↑