search

google-research / multinerf

A Code Release for Mip-NeRF 360, Ref-NeRF, and RawNeRF
Apache License 2.0
3.6k stars 341 forks source link

Key error in at fn_inv = inv_mapping[fn] in construct_ray_warps function #4

Closed Akhila-ar closed 2 years ago

Akhila-ar commented 2 years ago

I'm trying to run this codebase with the 360_v2 dataset provided. I am using one of the scenes from this data, and trying to train the model using train_360.sh script. What might be the issue here?

A slightly more detailed error log: File "/home/ubuntu/work/video_to_3d/multinerf/train.py", line 67, in main setup = train_utils.setup_model(config, key, dataset=dataset) File "/home/ubuntu/work/video_to_3d/multinerf/internal/train_utils.py", line 403, in setup_model model, variables = models.construct_model(rng, dummy_rays, config) File "/home/ubuntu/work/video_to_3d/multinerf/internal/models.py", line 331, in construct_model init_variables = model.init( File "/home/ubuntu/anaconda3/envs/multinerf/lib/python3.9/contextlib.py", line 79, in inner return func(*args, **kwds) File "/home/ubuntu/work/video_to_3d/multinerf/internal/models.py", line 124, in __call__ _, s_to_t = coord.construct_ray_warps(self.raydist_fn, rays.near, rays.far) File "/home/ubuntu/work/video_to_3d/multinerf/internal/coord.py", line 94, in construct_ray_warps fn_inv = inv_mapping[fn] KeyError: <function reciprocal at 0x7fd224ed4550>

jonbarron commented 2 years ago

Hi, sorry about that, and thanks for reporting this. I just pushed a change that should fix this, please sync and let me know if you continue having trouble.

yzslab commented 2 years ago

Hi, sorry about that, and thanks for reporting this. I just pushed a change that should fix this, please sync and let me know if you continue having trouble.

I met the same problem, and your latest push works now! Thanks!

Akhila-ar commented 2 years ago

It works now, thank you!