Cc-Hy / CMKD

Cross-Modality Knowledge Distillation Network for Monocular 3D Object Detection (ECCV 2022 Oral)
Apache License 2.0
107 stars 9 forks source link

Use of customised amount of training data #22

Closed carlvinson70 closed 1 year ago

carlvinson70 commented 1 year ago

Hi author, does the code support training with customised amounts of training data (to reproduce Figure 7's results in the paper)? If not, how do we go about this?

Cc-Hy commented 1 year ago

Hi, A very direct way to do this is to customize a split file just like this. You can put the chosen sample ids into it and generate a .pkl file and use it for training. Another way is to modify the dataset.py file, where you modify kitti_infos to adaptively choose training samples.