Eaphan / UPIDet

Unleash the Potential of Image Branch for Cross-modal 3D Object Detection [NeurIPS2023]
Apache License 2.0
54 stars 7 forks source link

about segmentation labels #6

Closed ksh11023 closed 1 year ago

ksh11023 commented 1 year ago

Hi, Thank you for sharing your work!

When generating *train.pkl using kitti_dataste.py, I don't think 'segmenation label' is included.

Can you give me some advise for fixing this issue?

Thank you.

Eaphan commented 1 year ago

You need to replace the *train.pkl files with those I provided. Please refer to readme.md for details.

"Especially, for the 2D auxiliary task of semantic segmentation, we used the instance segmentation annotations as provided in KINS dataset. We incorporate mask of instance segmentation in kitti_infos_train/val.pkl and kitti_dbinfos_train.pkl. Please download them in this link and replace the original files."