search

nerf2nerf / nerf2nerf

MIT License
138 stars 4 forks source link

about mipnerf360 training code #13

Open MiaApr opened 1 year ago

MiaApr commented 1 year ago

Hey, thanks for sharing your excellent work!
In your paper you said that for rendered scenes you trained a NeRF model using its classical architecture, while real-world scenes are reconstructed with mipNeRF360. I found the classic nerf code in other issues, and could you please also share your mipNeRF360 training code please? Thanks a lot!

lilygoli commented 1 year ago

Hi,

We had used a code very close to multiNeRF released code (in JAX). The checkpoints can be converted to be loaded with PyTorch. I will ask to see if we have the script for the conversion available to share with you and will keep you posted.

lilygoli commented 1 year ago

Following up on this, the JAX multiNeRF code was used to train the scenes, then the weights of the saved Jax model were mapped to their equivalent Torch format and loaded in the cod provided here. Finally this model was saved using TorchScript.