search

emecercelik / ssl-3d-detection

39 stars 7 forks source link

Clarification on training command and alternating training #5

Open richardkxu opened 1 year ago

richardkxu commented 1 year ago

Hi, thanks for the great work! I noticed only the training command for ssl-pointpillar on nus is provided. Can you provide the training and testing commands for the combination of ssl-pointpillar/centerpoint on kitti/nus? I am a little confused on which config file to use due to the naming.

I think the config file should be centerpoint_02pillar_second_secfpn_4x8_cyclic_20e_nus.py for the following script. The *20e_nus_det.py config does not exist.

https://github.com/emecercelik/ssl-3d-detection/blob/e605ad616278cdd7cb0c6cd5b8479c8c3921c158/mmdetection3d/mmdetection3d/tools/test_our_centerpoint_nus.sh#L1

Also, the provided training command is for pretraining on flow task and fine tune on 3d detection task once. Can you provide the script or detailed procedure for the "alternating training strategy" mentioned in the paper?

Thanks in advanced!

MingyuLiu1 commented 1 year ago

Hi richardkxu,

Sorry for the "ugly" codes XD. Let me explain the training step by step.

  1. We did not do experiment of CenterPoint on KITTI, cuz the official CenterPoint did not support it.
  2. For PointPillars on KITTI, we don't support it currently. Sorry for that.
  3. Yes, you are right, there is no *det.py for CenterPoint, this one is correct centerpoint_02pillar_second_secfpn_4x8_cyclic_20e_nus.py.
  4. For the alternating training, I will upload the checkpoint combination function soon, and explain how to do the alternating training.

Best, Mingyu

richardkxu commented 1 year ago

Thanks for the reply! So ssl-pointpillar/centerpoint does not support KITTI. Can you provide the training + testing commands for ssl-pointpillar/centerpoint on nus dataset? I am a little confused on which one to use due to naming.

Yeah, looking forward to the part on alternating training.

HenryJunW commented 1 year ago

Hello @richardkxu do you figure out the scene flow training script for centerpoint on nuScenes? Thanks!