Any plan for pytorch version

[TontonTremblay]

Hello,

I was wondering if there would be any plans for a pytorch version, or if you heard of someone working on one :P.

[TontonTremblay]

Also any plans for a docker image.

[krrish94]

@TontonTremblay I've reimplemented in PyTorch: https://github.com/krrish94/nerf-pytorch

[yenchenlin]

Hello @TontonTremblay and @krrish94 , I also have a PyTorch implementation: https://github.com/yenchenlin/nerf-pytorch

It reproduced the results at 400x400 resolution. Pre-trained models on 6 scenes are provided, more will be added tomorrow. All functions are tested and it runs ~1.3 times faster than the tensorflow implementation.

[kwea123]

My implementation in pytorch-lightning, faster, less VRAM required, and multi-gpu training is supported! Update: I also provide code to reconstruct colored mesh. Update: Tutorial, and many other features on Unity integration. Update: Live demo