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

I have tried to reproduce performance in paper #77

Closed magnetmoment closed 2 years ago

magnetmoment commented 2 years ago

Thanks for sharing the code! It's really a good job. I plan to take this as part of my undergraduate graduation project. But I couldn't get the same AP40 3D as what you mentioned.(93.75, 86.18, 83.50 and mine is 93.60 85.94 81.46)

Is it normal? Maybe It's a matter of batch size. I used one gpu and set batch size = 6 without any other changes in config.py.

magnetmoment commented 2 years ago

I set batch size = 4 and it can get closer mAP.