taichi-nerfs

收录理由

做神经辐射场研究的人往往要跟手写的CUDA内核较劲，这个项目给了一条更省心的路。它用Taichi语言承担繁重的计算，PyTorch负责训练流程，把instant-ngp的管线重新实现了一遍，代码比原版那套CUDA密集的参考实现好读也好改。开箱即用的脚本覆盖了合成NeRF和360_v2数据集，还支持用自己拍的视频走COLMAP重建场景。训练完会弹出交互式界面，官方给出的性能数据也实在——在RTX 3090上跑Lego场景，约208秒就能达到35.0的PSNR。对于做三维重建和新视角合成的快速实验来说，这个工具相当顺手。另外它还能作为stable-dreamfusion的后端，支持文本生成三维模型，用途不限于纯重建。