how to generate kitti style train.txt file

tusen-ai / SST

Code for a series of work in LiDAR perception, including SST (CVPR 22), FSD (NeurIPS 22), FSD++ (TPAMI 23), FSDv2, and CTRL (ICCV 23, oral).

Apache License 2.0

765 stars 96 forks source link

how to generate kitti style train.txt file #168

Closed SakuraRiven closed 8 months ago

SakuraRiven commented 9 months ago

Hi, we try to use python tools/create_data.py but found that it needs txt files such as ImageSets/train.txt in kitti_format dir.

How could we generate these txt files so that we can reproduce the CTRL work?

Thanks for your kind help : )

SakuraRiven commented 9 months ago

Our method: collect the idxs from idx2contextname.pkl and assign idxs to train/val/test in terms of their corresponding context_names. In this way, the idxs could be same as the official waymo split.

Abyssaledge commented 9 months ago

Sorry for the late reply. We have been quite busy lately. Your solution seems nice, did it work? I can upload these files if you need.

SakuraRiven commented 8 months ago

@Abyssaledge It works well and we have reproduced the method successfully : )