search

muzi2045 / second_TANET.pytorch

MIT License
5 stars 2 forks source link

PointPillars to ONNX #1

Closed hz3014 closed 4 years ago

hz3014 commented 4 years ago

Hey, I was wondering how did you export your pytorch modle to onnx? Did you use SmallMunich code to export it to pfe.onnx and rpn.onnx or did you export the whole modle into one onnx file?

muzi2045 commented 4 years ago

the best way to export pointpillars to onnx is split it into two part. because the pillars scatter have to be writen in CUDA kernel, or you can write the scatter layer into trt plugin. I'm not using other onnx export scripts, because here is some place in network I changed for more faster inference speed, so I write it myself.

hz3014 commented 4 years ago

eca

I have been working on exporting Second(NuScenes) to onnx for weeks and I am very confused. It seems like everything in voxelnet.py need to be modified. Would you please show some guidelines for exporting onnx?