search

HaohaoNJU / CenterPoint

TensorRT deployment for CenterPoint Lidar Detection Model.
MIT License
270 stars 56 forks source link

ONNX model incorrect! #14

Closed zhi-yuan-song closed 2 years ago

zhi-yuan-song commented 2 years ago

This your pfe onnx model visual dispaly. pfe_baseline32000 onnx This my pfe onnx model visual dispaly image I think my model is not correct. Why did my action produce an incorrect model?

HaohaoNJU commented 2 years ago

You need to check the code, there must be some modification within the computation graph, but I think this wouldn't affect the performance , you should replace the pfe output tensor name from "47" to "45", see here https://github.com/Abraham423/CenterPoint/blob/4d70d09726050b682d27525aa65743ff106c6b9a/src/samplecenterpoint.cpp#L111

zhi-yuan-song commented 2 years ago

I have made changes to the c++ code and tested it,the results is correct! Thanks~