CenterFusion icon indicating copy to clipboard operation
CenterFusion copied to clipboard

what is the details when produce the cennternet_baseline_e170.pth model file?

Open ryontang opened this issue 1 year ago • 2 comments

hi thanks for your work, and I decided to train the CenterFusion by local dataset but not nuScenes.

As you said from README: The centernet_baseline_e170 model is obtained by starting from the original CenterNet 3D detection model nuScenes_3Ddetection_e140 and training the velocity and attributes heads for 30 epochs.

I understand it as I need to use the original CenterNet to trian to get the first model file(nuScenes_3Ddetection_e140), and then I use the CenterFusion code to trian the velocity and attributes heads for 30 epochs, so that I could get the pre-train model-centernet_baseline_e170.

so do I understand it correctly?

ryontang avatar Mar 07 '23 10:03 ryontang

where can i get this pretrained model center_baseline_e170?

Manishnayak234 avatar Jan 05 '24 15:01 Manishnayak234