DreamFusion:基于 2D 扩散先验的文本到 3D 生成

FreeGuideOnline 最新 2026-06-25

bash pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118 pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch pip install diffusers transformers accelerate


### 使用官方实现运行 DreamFusion

Google Research 提供了 DreamFusion 的官方实现(基于 JAX),社区也开发了多个 PyTorch 复现版本,如 `threestudio` 和 `stable-dreamfusion`。以下以 `stable-dreamfusion` 为例:

1. **克隆仓库并安装**
   ```bash
   git clone https://github.com/ashawkey/stable-dreamfusion.git
   cd stable-dreamfusion
   pip install -r requirements.txt
  1. 下载预训练扩散模型权重 DreamFusion 默认需要 Stable Diffusion 2.1 权重,可从 Hugging Face 自动下载,或手动放置于 pretrained/ 目录。

  2. 运行生成命令

    python main.py --text "A cute rabbit, detailed, 4K" --workspace rabbit_trial -O