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
-
下载预训练扩散模型权重 DreamFusion 默认需要 Stable Diffusion 2.1 权重,可从 Hugging Face 自动下载,或手动放置于
pretrained/目录。 -
运行生成命令
python main.py --text "A cute rabbit, detailed, 4K" --workspace rabbit_trial -O