nerfstudio

nerfstudio

Nerfstudio，一个用于NeRF开发的模块化PyTorch框架。框架中用于实现基于NeRF的方法的组件即插即用，使得研究人员和相关从业者可以轻松地将NeRF集成到自己的项目中。框架的模块化设计支持实时可视化工具，导入用户真实世界捕获的数据集外（in-the-wild）数据，以及导出为视频，点云和网格表示的工具。

安装

官网安装方式可能会导致tiny-cuda-nn安装失败

conda create --name nerfstudio -y python=3.8
conda activate nerfstudio

conda install pytorch==2.1.2 torchvision==0.16.2 pytorch-cuda=11.8 -c pytorch -c nvidia
pip install ninja git+https://gitee.com/Xiaoshier2021/tiny-cuda-nn/#subdirectory=bindings/torch
conda install -c "nvidia/label/cuda-11.8.0" cuda-toolkit

git clone https://github.com/nerfstudio-project/nerfstudio.git
cd nerfstudio
pip install -e .

前处理数据

ns-process-data images --data ~/data_nerf/images/ --output-dir ~/data_nerf/out/

训练

ns-train nerfacto --data ~/data_nerf/out/ --vis viewer+tensorboard

可视化工具：http://localhost:7007/

评测

ns-eval --load-config outputs/out/nerfacto/2024-09-20_100756/config.yml --output_path outputs/out/nerfacto/result.json --render-output-path outputs/out/nerfacto/render