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

Shap-E:生成条件式3D隐函数

Shap-E: Generating Conditional 3D Implicit Functions

OpenAI

Heewoo Jun, Alex Nichol

AI总结:

本研究提出3D资产条件生成模型Shap-E,通过编码器+条件扩散模型两阶段训练,直接生成可渲染为带纹理网格或神经辐射场的隐函数参数,生成速度快、质量优于同类模型Point-E。

摘要:

我们提出Shap-E,一种面向3D资产的条件生成模型。与近期生成单一输出表示的3D生成模型研究不同,Shap-E直接生成隐函数的参数,这些参数既可渲染为带纹理的网格,也可渲染为神经辐射场(NeRF)。我们分两个阶段训练Shap-E:第一阶段训练一个编码器,将3D资产确定性地映射为隐函数的参数;第二阶段在编码器的输出上训练条件扩散模型。当在大规模3D与文本配对数据集上训练后,所得模型能在数秒内生成复杂多样的3D资产。与基于点云的显式生成模型Point-E相比,尽管Shap-E建模的是维度更高、支持多表示的输出空间,但它收敛更快,样本质量相当甚至更优。我们已在https://github.com/openai/shap-e发布模型权重、推理代码和样本。

英文摘要:

We present Shap-E, a conditional generative model for 3D assets. Unlike recent work on 3D generative models which produce a single output representation, Shap-E directly generates the parameters of implicit functions that can be rendered as both textured meshes and neural radiance fields. We train Shap-E in two stages: first, we train an encoder that deterministically maps 3D assets into the parameters of an implicit function; second, we train a conditional diffusion model on outputs of the encoder. When trained on a large dataset of paired 3D and text data, our resulting models are capable of generating complex and diverse 3D assets in a matter of seconds. When compared to Point-E, an explicit generative model over point clouds, Shap-E converges faster and reaches comparable or better sample quality despite modeling a higher-dimensional, multi-representation output space. We release model weights, inference code, and samples at https://github.com/openai/shap-e.

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

↑