GenFusion项目教程

1. 项目目录结构及介绍

GenFusion项目的目录结构如下：

GenFusion/
├── assets/               # 存放项目所需的资源文件
├── diffusion_ckpt/       # 存放训练好的模型权重文件
├── environment.yml       # Conda环境配置文件
├── LICENCE               # 项目许可证文件
├── README.md             # 项目说明文件
├── Reconstruction/       # 3D重建相关代码和脚本
├── GenerationModel/      # 生成模型相关代码和脚本
└── genfusion_scripts/    # 通用脚本文件

• assets/：包含项目所需的资源文件，如数据集等。
• diffusion_ckpt/：存放训练好的模型权重文件，用于3D重建和生成模型。
• environment.yml：Conda环境配置文件，用于创建项目所需的环境。
• LICENCE：项目许可证文件，本项目采用MIT协议。
• README.md：项目说明文件，详细介绍项目背景、功能和用法。
• Reconstruction/：3D重建相关代码和脚本，包括重建、稀疏视图重建和场景补全等。
• GenerationModel/：生成模型相关代码和脚本，用于训练和测试生成模型。
• genfusion_scripts/：通用脚本文件，包含一些项目通用的脚本。

2. 项目的启动文件介绍

项目的启动文件主要有两个：train.py 和 batch_*.py 脚本。

• train.py：项目的主启动文件，用于执行3D重建和生成模型的训练、测试和推理等任务。根据不同的命令行参数，可以执行不同的功能。

• batch_*.py：批量执行任务的相关脚本。例如 batch_ours.py 用于批量执行3D重建任务，batch_sparse.py 用于批量执行稀疏视图重建任务。

3. 项目的配置文件介绍

项目的配置文件主要有两个：environment.yml 和 *.yaml 文件。

• environment.yml：Conda环境配置文件，用于创建项目所需的环境。文件内容如下：

name: genfusion
channels:
  - pytorch
  - conda-forge
dependencies:
  - python=3.8
  - torch=1.10.0+cu113
  - torchvision=0.11.1+cu113
  - numpy
  - matplotlib
  - scikit-image
  - opencv-python
  - scipy
  - tensorboard

• *.yaml：生成模型和3D重建相关的配置文件。例如 generation_infer.yaml 用于配置生成模型推理时的参数。文件内容如下：

model:
  name: "GenFusion"
  type: "VideoGenFusion"
  generator:
    name: "DynamicCrafter"
    type: "DynamicCrafter"
    ... # 更多配置参数

通过修改这些配置文件，用户可以根据自己的需求调整项目运行时的环境和参数。