avishkarsaha / translating-images-into-maps

Official PyTorch code for 'Translating Images Into Maps' ICRA 2022 (Outstanding Paper Award)
Other
408 stars 49 forks source link

Problem running Pretrained Weights #39

Open hocherie opened 7 months ago

hocherie commented 7 months ago

Hi @avishkarsaha , thank you for the great work.

I am trying to run inference with your pretrained weights, however am having trouble.

The pretrained weights seem to be using (according to config.txt) PyrOccNet_S_0914_old_rep100x100_out100x100.

However, this model does not exist in network.py. Can you please share the updated network.py? And give more guidance on how to infer and visualize? Thank you!

When I try using the default model_name PyrOccTranDetr_S_0904_old_rep100x100_out100x100, there is a huge mismatch between available keys

The resulting predictions does not look representative of the results

image image