GeoDream 项目使用教程

GeoDream 项目使用教程

GeoDream GeoDream: Disentangling 2D and Geometric Priors for High-Fidelity and Consistent 3D Generation 项目地址: https://gitcode.com/gh_mirrors/ge/GeoDream

1. 项目目录结构及介绍

GeoDream 项目的目录结构如下：

GeoDream/
├── assets/
├── configs/
├── docker/
├── docs/
├── extern/
├── load/
├── mv-diffusion/
├── scripts/
├── threestudio/
├── 2dplayground.ipynb
├── CHANGELOG.md
├── DOCUMENTATION.md
├── LICENSE
├── README.md
├── export_mesh.sh
├── gradio_app.py
├── launch.py
├── mesh-finetuning-geo.sh
├── mesh-finetuning-texture.sh
├── neus-train.sh
├── requirements.txt
├── test_rendering.sh
└── threestudio.ipynb

目录介绍

• assets/: 存放项目相关的静态资源文件。
• configs/: 存放项目的配置文件。
• docker/: 存放与 Docker 相关的文件和脚本。
• docs/: 存放项目的文档文件。
• extern/: 存放外部依赖或插件。
• load/: 存放加载模块的文件。
• mv-diffusion/: 存放多视图扩散模型的相关文件。
• scripts/: 存放项目的脚本文件。
• threestudio/: 存放与 ThreeStudio 扩展相关的文件。
• 2dplayground.ipynb: 2D 游戏场的 Jupyter Notebook 文件。
• CHANGELOG.md: 项目更新日志。
• DOCUMENTATION.md: 项目文档。
• LICENSE: 项目许可证。
• README.md: 项目介绍和使用说明。
• export_mesh.sh: 导出网格的脚本。
• gradio_app.py: Gradio 应用的 Python 文件。
• launch.py: 项目启动文件。
• mesh-finetuning-geo.sh: 网格几何微调的脚本。
• mesh-finetuning-texture.sh: 网格纹理微调的脚本。
• neus-train.sh: NeuS 训练的脚本。
• requirements.txt: 项目依赖的 Python 包列表。
• test_rendering.sh: 渲染测试的脚本。
• threestudio.ipynb: ThreeStudio 的 Jupyter Notebook 文件。

2. 项目启动文件介绍

项目的启动文件是 launch.py。该文件负责启动项目的训练和生成过程。以下是启动文件的基本使用方法：

python launch.py --config configs/geodream-neus.yaml --train --gpu 0

参数说明

• --config: 指定配置文件的路径。
• --train: 启动训练模式。
• --gpu: 指定使用的 GPU 设备号。

3. 项目的配置文件介绍

配置文件存放在 configs/ 目录下，主要的配置文件包括：

• geodream-neus.yaml: 用于 NeuS 训练的配置文件。
• geodream-dmtet-geometry.yaml: 用于 DMTet 几何优化的配置文件。
• geodream-dmtet-texture.yaml: 用于 DMTet 纹理优化的配置文件。

配置文件示例

system:
  prompt_processor:
    prompt: "an astronaut riding a horse"
  geometry:
    init_volume_path: "mv-diffusion/volume/An_astronaut_riding_a_horse/con_volume_lod_150.pth"
  renderer:
    context_type: cuda
  geometry_convert_override:
    isosurface_threshold: 0.0

配置项说明

• prompt_processor: 处理输入提示的配置。
• geometry: 几何体初始化路径的配置。
• renderer: 渲染器上下文类型的配置。
• geometry_convert_override: 几何体转换覆盖的配置。

通过以上配置文件，可以灵活地调整项目的训练和生成参数，以满足不同的需求。

GeoDream GeoDream: Disentangling 2D and Geometric Priors for High-Fidelity and Consistent 3D Generation 项目地址: https://gitcode.com/gh_mirrors/ge/GeoDream