HaohaoNJU/CenterPoint

ONNX model incorrect!

Closed this issue · 2 comments

zhi-yuan-song commented

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

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

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