HDR-Plenoxels 项目使用教程

1. 项目目录结构及介绍

HDR-Plenoxels/
├── build/
│   ├── opt/
│   │   ├── hdr_opt.py
│   │   ├── shell/
│   │   │   ├── syn/
│   │   │   │   ├── train_mid.sh
│   │   │   │   ├── train_mix.sh
│   │   │   │   ├── train_tone.sh
│   │   │   ├── real/
│   │   │   │   ├── train_mid.sh
│   │   │   │   ├── train_mix.sh
│   │   │   │   ├── train_tone.sh
│   │   ├── render/
│   │   │   ├── render_hdr.sh
│   │   │   ├── render_ldr.sh
│   │   ├── hdr_calc_metrics.py
├── svox2/
│   ├── __init__.py
│   ├── ... (其他相关文件)
├── test/
│   ├── ... (测试相关文件)
├── LICENSE
├── README.md
├── environment.yml
├── manual_install.sh
├── setup.py

目录结构说明

• build/: 包含项目的构建和优化相关文件。
  • opt/: 包含训练和渲染的脚本。
    • hdr_opt.py: 训练单个场景的主要脚本。
    • shell/: 包含用于训练和渲染的Shell脚本。
      • syn/: 包含用于合成数据集的训练脚本。
      • real/: 包含用于真实数据集的训练脚本。
    • render/: 包含用于渲染HDR和LDR辐射场的脚本。
    • hdr_calc_metrics.py: 用于计算评估指标的脚本。
• svox2/: 包含项目的主要代码库，包括CUDA扩展和其他相关文件。
• test/: 包含项目的测试文件。
• LICENSE: 项目的许可证文件。
• README.md: 项目的介绍和使用说明。
• environment.yml: 用于创建虚拟环境的配置文件。
• manual_install.sh: 手动安装脚本。
• setup.py: 项目的安装脚本。

2. 项目的启动文件介绍

hdr_opt.py

hdr_opt.py 是项目的主要启动文件，用于训练单个场景的HDR辐射场。该脚本位于 build/opt/ 目录下。

使用方法

python hdr_opt.py --config path/to/config.yaml

其他启动脚本

• train_mid.sh, train_mix.sh, train_tone.sh: 这些脚本位于 build/opt/shell/syn/ 和 build/opt/shell/real/ 目录下，分别用于训练不同类型的数据集。

3. 项目的配置文件介绍

environment.yml

environment.yml 是用于创建虚拟环境的配置文件。通过该文件，可以快速创建一个包含所有依赖项的虚拟环境。

使用方法

conda env create -f environment.yml
conda activate plenoxel

setup.py

setup.py 是项目的安装脚本，用于安装项目的主要库和依赖项。

使用方法

pip install .

manual_install.sh

manual_install.sh 是一个手动安装脚本，用于在CUDA工具包较旧的情况下安装CUB库。

使用方法

./manual_install.sh

通过以上步骤，您可以顺利地启动和配置HDR-Plenoxels项目，并开始进行HDR辐射场的训练和渲染。