wzzheng / TPVFormer

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

tpv04 setting Lidar Segmentation Pretrain #33

Open gaopeng91 opened 1 year ago

gaopeng91 commented 1 year ago

Hi, thanks for the great job.

could you release the weights for Lidar Segmentation in tpv04 setting? tpv10 setting is too large for 3090.

huang-yh commented 1 year ago

Sorry, but the setting we used to train tpvformer04 for lidar segmentation had TPV resolution of only 1001008, which yielded poor performance. So we are not planning to release the weight. Btw, you could also try training with other configs of tpvformer10 for lidarseg, i.e. dim64 or dim96, because they should fit into 3090 and also achieve comparable performance.