search

mrnabati / CenterFusion

CenterFusion: Center-based Radar and Camera Fusion for 3D Object Detection
MIT License
546 stars 140 forks source link

change model #29

Closed AHappyFlyBird closed 3 years ago

AHappyFlyBird commented 3 years ago

Hi! friend，I need your help. When I try test.sh , I change the model from centerfusion modle to nuscens_3Ddetection_e140 model,like the picture: image But the result like this image ,the error is too big,Is this a normal result?OR, Is there any parameters that I need to change or add

mrnabati commented 3 years ago

Hi. If you want to get the exact CenterNet results you should run the CenterNet code. CenterFusion has many modifications and might not give you the correct CenterNet results even when modifying the test script options like what you've done. But if you want to recreate the baseline CenterNet that we compare to in the paper, you need to use the pre-trained centernet_baseline_e170.pth model, which trains the nuScenes_3Ddetection_e140.pth model for 30 more epochs to train the velocity and attributes heads.