PSP-HDRI$+$：用于以人为中心计算机视觉模型预训练的合成数据集生成器
计算机视觉与模式识别
2022-07-12 v1 人工智能
数据库
图形学
机器学习
摘要
我们引入一个新的合成数据生成器 PSP-HDRI，它被证明是 ImageNet 及其他大规模合成数据对应物的更优预训练替代方案。我们证明，使用我们的合成数据进行预训练将产生更通用的模型，即使在分布外（OOD）集上测试时也优于替代方案。此外，利用由现成模型架构上的人体关键点估计指标引导的消融研究，我们展示了如何操控我们的合成数据生成器以进一步提升模型性能。
引用
@article{arxiv.2207.05025,
title = {PSP-HDRI$+$: A Synthetic Dataset Generator for Pre-Training of Human-Centric Computer Vision Models},
author = {Salehe Erfanian Ebadi and Saurav Dhakad and Sanjay Vishwakarma and Chunpu Wang and You-Cyuan Jhang and Maciek Chociej and Adam Crespi and Alex Thaman and Sujoy Ganguly},
journal= {arXiv preprint arXiv:2207.05025},
year = {2022}
}
备注
PSP-HDRI$+$ template Unity environment, benchmark binaries, and source code will be made available at: https://github.com/Unity-Technologies/PeopleSansPeople