search

V2AI / Det3D

World's first general purpose 3D object detection codebse.
https://arxiv.org/abs/1908.09492
Apache License 2.0
1.48k stars 299 forks source link

PointPillars speed #89

Closed Nuri-benbarka closed 4 years ago

Nuri-benbarka commented 4 years ago

I trained PointPillars with the config file you provided on my RTX 2060 GPU and achieved a speed of 21.4Hz. In the publication of PointPillars, they reported a speed of 62Hz on their GTX1080ti which should be ~50% faster than mine, so I was expecting a speed aroand 40Hz. In general, is there a speed loss when using this framework?

Screenshot from 2020-03-26 12-41-18

poodarchu commented 4 years ago

I think they only take forward time into consideration. but infact there're also data preparation and post processing time.

jhultman commented 4 years ago

Hi @Nuri-benbarka, the PointPillars authors used TensorRT to reach 62 hz (42 hz without TensorRT), including all pre- and postprocessing.

tensorrt

Nuri-benbarka commented 4 years ago

Many thanks, @jhultman. Now the runtimes make sense to me.