KeyError: 0
Closed this issue · 3 comments
Thanks for your fantastic work!
I am trying to use Mars to train the kitti-0000, but I got this error: KeyError:0
File "/data/code/test-mars/mars/models/car_nerf.py", line 206, in inference_without_render
self.car_latents[int(obj_id)].view(1, -1).to(obj_id.device),
KeyError: 0
I use the latent_codes_car_van_truck.pt
,but this pt file only provides 10 obj latents, like this
But kitti-0000 includes [array ([0.]), array ([3.]), array ([5.]), array ([4.]), array ([6.]), array ([9.])] in this scene
So how can I train the mars in kitti-0000?
Looking forward to your reply!
Hi! @heiheishuang
Thanks for your reply. We haven't provided the latent code for kitti-0000, you can train with nerfacto or randomize a latent code with the corresponding object id.
Thanks for you reply!!!
May I ask if you will can provide the PSNR and other metrics for kitti-0000?
Looking forward to your reply!
Hi! @heiheishuang
Sorry that we haven't done any experiments on kitti-0000.