HashNeRF-pytorch

收录理由

想弄明白Instant-NGP为何能把神经辐射场训练得这么快的研究者，可以顺着这个纯PyTorch移植版把代码走一遍，完全不需要碰原来的CUDA和C++实现。方法核心的多分辨率哈希编码在这里被组织得清晰且自包含，项目又建立在NeRF-pytorch之上，熟悉那个代码库的人一眼就能找到熟悉的脉络。用一块1050Ti这种入门级显卡训练椅子场景，大约五千次迭代、十来分钟，就能看到相当锐利的渲染结果。代码里还额外支持全变分损失和稀疏性损失，想平滑嵌入或清理几何结构，改个参数就行。由于全程都是Python，你可以直接修改网络结构，逐步追踪每一部分到底起了什么作用。对于想在NeRF方向上做实验、而不是花时间跟原生代码较劲的人来说，这是一个很顺手的起点。