Acquiring labeled datasets for 3D human mesh estimation is challenging due to depth ambiguities and the inherent difficulty of annotating 3D geometry from monocular images. Existing datasets are either real, with manually annotated 3D geometry and limited scale, or synthetic, rendered from 3D engines that provide precise labels but suffer from limited photorealism, low diversity, and high production costs. In this work, we explore a third path: generated data. We introduce PoseDreamer, a novel pipeline that leverages diffusion models to generate large-scale synthetic datasets with 3D mesh annotations. Our approach combines controllable image generation with Direct Preference Optimization for control alignment, curriculum-based hard sample mining, and multi-stage quality filtering. Together, these components naturally maintain correspondence between 3D labels and generated images, while prioritizing challenging samples to maximize dataset utility. Using PoseDreamer, we generate more than 500,000 high-quality synthetic samples, achieving a 76% improvement in image-quality metrics compared to rendering-based datasets. Models trained on PoseDreamer achieve performance comparable to or superior to those trained on real-world and traditional synthetic datasets. In addition, combining PoseDreamer with synthetic datasets results in better performance than combining real-world and synthetic datasets, demonstrating the complementary nature of our dataset. We will release the full dataset and generation code.
翻译:获取用于三维人体网格估计的标注数据集极具挑战性,其原因在于深度歧义性以及从单目图像标注三维几何结构的固有困难。现有数据集要么是真实数据集(人工标注三维几何结构且规模有限),要么是合成数据集(通过三维引擎渲染生成并提供精确标签,但存在逼真度不足、多样性低以及生产成本高的问题)。本文探索了第三条路径:生成数据。我们提出PoseDreamer,一种创新性管线,利用扩散模型生成大规模带有三维网格标注的合成数据集。本方法将可控图像生成与用于控制对齐的直接偏好优化、基于课程学习的困难样本挖掘以及多阶段质量过滤相结合。这些组件自然地保持了三维标签与生成图像之间的对应关系,同时优先处理挑战性样本以最大化数据集效用。基于PoseDreamer,我们生成了超过50万张高质量合成样本,与基于渲染的数据集相比,图像质量指标提升了76%。使用PoseDreamer训练的模型在性能上可媲美甚至超越基于真实世界和传统合成数据集训练的模型。此外,将PoseDreamer与合成数据集结合使用,其表现优于真实世界数据集与合成数据集的组合,充分展示了我们数据集的互补特性。我们将完整发布该数据集及生成代码。