search

TuSimple / centerformer

Implementation for CenterFormer: Center-based Transformer for 3D Object Detection (ECCV 2022)
MIT License
293 stars 28 forks source link

Why only selected the 0-th data in "example["ind"][0]", I think there are 6 task head? #5

Open rockywind opened 1 year ago

rockywind commented 1 year ago

Hi, I'm confused about that why only selected the 0-th data in "example["ind"][0]" in the line , I think there are 6 task head?

edwardzhou130 commented 1 year ago

We use only 1 task head on Waymo Open Dataset. https://github.com/TuSimple/centerformer/blob/0408b29fa728950baead73354e21240174ce5b6b/configs/waymo/voxelnet/waymo_centerformer.py#L6-L8