Hi,thanks for your work! Here I’ve encountered some problems when run your code. I think there are some mistakes in reprojector function in match.py file. looking forward to your update! Thank you!
Cannot divide evenly the sizes of shapes (3, 16) and (14400, 16)
File "/home/bran/GeCoNeRF-main/train.py", line 488, in
app.run(main)
File "/home/bran/GeCoNeRF-main/train.py", line 330, in main
w_grad, outputs, keys = reg_utils.reg_train_step(render_train_fn_, reg_model, dataset_type, 120, reg_type, keys[0], state, reg_params, reg_batch, alpha_batch, step)
File "/home/bran/GeCoNeRF-main/internal/reg_utils.py", line 155, in reg_train_step
(loss , outputs), grad = jax.value_and_grad(warp_loss_fn, has_aux = True)(jax.device_get(jax.tree_map(lambda x:x[0], state)).optimizer.target)
File "/home/bran/GeCoNeRF-main/internal/reg_utils.py", line 93, in warp_loss_fn
data_type = data_type
File "/home/bran/GeCoNeRF-main/internal/models.py", line 377, in render_image
rng, data_type, eval = eval)
File "/home/bran/GeCoNeRF-main/internal/match.py", line 127, in reprojector
pose = pose.reshape(rays.pose.shape),

The above exception was the direct cause of the following exception:

File "/home/bran/GeCoNeRF-main/internal/match.py", line 127, in reprojector
pose = pose.reshape(rays.pose.shape),
File "/home/bran/GeCoNeRF-main/internal/models.py", line 377, in render_image
rng, data_type, eval = eval)
File "/home/bran/GeCoNeRF-main/internal/reg_utils.py", line 93, in warp_loss_fn
data_type = data_type
File "/home/bran/GeCoNeRF-main/internal/reg_utils.py", line 155, in reg_train_step
(loss , outputs), grad = jax.value_and_grad(warp_loss_fn, has_aux = True)(jax.device_get(jax.tree_map(lambda x:x[0], state)).optimizer.target)
File "/home/bran/GeCoNeRF-main/train.py", line 330, in main
w_grad, outputs, keys = reg_utils.reg_train_step(render_train_fn_, reg_model, dataset_type, 120, reg_type, keys[0], state, reg_params, reg_batch, alpha_batch, step)
File "/home/bran/GeCoNeRF-main/train.py", line 488, in
app.run(main)