OPEN-AIR-SUN/mars

KeyError: 0

Closed this issue · 3 comments

heiheishuang commented

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
latents
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!

JiantengChen commented

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.

heiheishuang commented

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!

JiantengChen commented

Hi! @heiheishuang

Sorry that we haven't done any experiments on kitti-0000.