search

wzzheng / TPVFormer

[CVPR 2023] An academic alternative to Tesla's occupancy network for autonomous driving.
https://wzzheng.net/TPVFormer/
Apache License 2.0
1.16k stars 105 forks source link

What is config for the released tpv10_lidarseg_v2.pth? #53

Open KingsleyLiu-NV opened 11 months ago

KingsleyLiu-NV commented 11 months ago

None of tpv_lidarseg.py, tpv_lidarseg_dim64.py, tpv_lidarseg_dim96.py works.

Traceback (most recent call last):
  File "visualization/vis_frame.py", line 231, in <module>
    print(my_model.load_state_dict(revise_ckpt(ckpt)))
  File "/root/miniconda3/envs/py37t181/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1224, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for TPVFormer:
        Unexpected key(s) in state_dict: "tpv_head.encoder.ref_2d_hw", "tpv_head.encoder.ref_2d_zh", "tpv_head.encoder.ref_2d_wz".