Problem running Pretrained Weights

[hocherie]

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