Nerfstudio的调试方法

Nerfstudio的调试方法

Nerfstudio是一个基于tyro的CLI的渲染工具包. 其所需训练,验证,可视化等功能均在终端中进行. 基于命令固然轻量易于管理, 但是却不利于开发者在开发的过程中进行调试. 本文以我个人的经验给大家一个调试的思路.

Nerfstudio的下载及安装方式详见: Nerfstudio

1 在随意位置创建test.py文件

即使以源码通过pip install -e .安装Nerfstudio, 我们不希望在其原始仓库中进行改动. 我们在任意位置创建工作空间并新建文件test.py进行测试.

2 写入train程序

我们将nerfstudio包中nerfstudio/scripts/train.py位置中的代码完全复制粘贴到test.py中.

3 调试程序

• 引用nerfstudio中所有method

from nerfstudio.configs.method_configs import all_methods, all_descriptions

• 以nerfacto方法为例

if __name__ == '__main__':
    from pathlib import Path
    methods = all_methods['nerfacto']
    des = all_descriptions['nerfacto']
    methods.data = Path(f'{Your data path}')
    main(methods)

• 断点调试即可
  如果使用VSCode, 需设置launch文件为: "justMyCode": false

4 其他

• 如果对ipdb熟悉的朋友, 也可以选择在终端使用ipdb进行调试.
• 更具体的, 大家可以在

from nerfstudio.configs.method_configs import AnnotatedBaseConfigUnion

中找到或者写自己的方法具体的配置文件, 这样在调试的过程中就不用一一指定配置了.

如果其它问题欢迎留言或联系E-mail,大家共同进步.