search

tianweiy / CenterPoint-KITTI

Apache License 2.0
180 stars 44 forks source link

[IMPROVEMENT] Target Assignment is Slow... #24

Open tianweiy opened 3 years ago

tianweiy commented 3 years ago

The current target assignment for the CenterPoint head is super slow (> 50% of the total runtime). We need to figure out some ways to improve this

HuangVictorAuto commented 2 years ago

Hey, Tianwei. You mean the target assignment here based on Openpcdet is slow and needs improments. Is the target assignment in your official repo also slow and need improvements?

tianweiy commented 2 years ago

for openpcdet + centerpoint it is slow (other algorithm in openpcdet are relatively faster). The target assignment in my official repo is super fast as it is performed in the dataloader with multiprocessing.