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

OctFusion:用于3D形状生成的基于八叉树的扩散模型

OctFusion: Octree-based Diffusion Models for 3D Shape Generation

Peking University(北京大学);Tsinghua University(清华大学);VAST(中国科学院软件研究所)

Bojun Xiong, Si-Tong Wei, Xin-Yang Zheng, Yan-Pei Cao, Zhouhui Lian, Peng-Shuai Wang

AI总结:

提出OctFusion,利用八叉树潜在表示与统一多尺度U-Net扩散模型,实现快速、任意分辨率且保证流形网格的高质量3D形状生成,并支持颜色及多条件生成。

摘要:

扩散模型已成为3D生成中的一种流行方法。然而,扩散模型要高效生成多样且高质量的3D形状仍具挑战。本文提出OctFusion,可在单块Nvidia 4090 GPU上用2.5秒生成任意分辨率的3D形状,并且提取出的网格保证连续且为流形。OctFusion的关键组成是基于八叉树的潜在表示及配套扩散模型。该表示结合了隐式神经表示与显式空间八叉树的优势,并通过基于八叉树的变分自编码器学习。所提出的扩散模型是统一的多尺度U-Net,可在不同八叉树层级间共享权重和计算,避免了广泛使用的级联扩散方案的复杂性。我们在ShapeNet和Objaverse数据集上验证了OctFusion的有效性,并在形状生成任务上取得最先进性能。我们还通过生成用于带纹理网格生成的高质量颜色场,以及以文本提示、草图或类别标签为条件的高质量3D形状,证明OctFusion具有可扩展性和灵活性。代码和预训练模型位于https://github.com/octree-nn/octfusion。

英文摘要:

Diffusion models have emerged as a popular method for 3D generation. However, it is still challenging for diffusion models to efficiently generate diverse and high-quality 3D shapes. In this paper, we introduce OctFusion, which can generate 3D shapes with arbitrary resolutions in 2.5 seconds on a single Nvidia 4090 GPU, and the extracted meshes are guaranteed to be continuous and manifold. The key components of OctFusion are the octree-based latent representation and the accompanying diffusion models. The representation combines the benefits of both implicit neural representations and explicit spatial octrees and is learned with an octree-based variational autoencoder. The proposed diffusion model is a unified multi-scale U-Net that enables weights and computation sharing across different octree levels and avoids the complexity of widely used cascaded diffusion schemes. We verify the effectiveness of OctFusion on the ShapeNet and Objaverse datasets and achieve state-of-the-art performances on shape generation tasks. We demonstrate that OctFusion is extendable and flexible by generating high-quality color fields for textured mesh generation and high-quality 3D shapes conditioned on text prompts, sketches, or category labels. Our code and pre-trained models are available at https://github.com/octree-nn/octfusion.

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

↑