search

Vegeta2020 / SE-SSD

SE-SSD: Self-Ensembling Single-Stage Object Detector From Point Cloud, CVPR 2021.
Apache License 2.0
811 stars 128 forks source link

original_pp_mghead_syncbn_kitti.py(Point_Pillars) #31

Closed sunwonHong closed 3 years ago

sunwonHong commented 3 years ago

original_pp_mghead_syncbn_kitti.py is for pointpillars Net, right?

To use pointpillars, all i have to do is change this on train.py? before : parser.add_argument("--config", default='../examples/second/configs/config.py', help="train config file path") after : parser.add_argument("--config", default='../examples/point_pillars/configs/original_pp_mghead_syncbn_kitti.py', help="train config file path")

I think additional process must be needed. Could you let know me how i can use pointpillars? What should i change?

Vegeta2020 commented 3 years ago

I haven't done training with pointpillars, this file is just saved as in Det3D. I guess you need to check the pointpillar network details to be consistent with the input and output data processing.

sunwonHong commented 3 years ago

OK. Thank you for replying!