bevfusion icon indicating copy to clipboard operation
bevfusion copied to clipboard

How to convert a trained model to onnx

Open YueSun0609 opened this issue 1 year ago • 2 comments

Thank you for your kind release of the codes, I have successfully applied it to my own data and got good performace. But may I ask how to export a trained model to onnx format? In tools/export.py, you gave an example but missing "forward_test" function. I tried to write one, but when I run torch.onnx.export() function, it meets an error like "RuntimeError: Tracer cannot infer type of [{'boxes_3d': LiDARInstance3DBoxes", could you give me some clues on how to solve this issue? Thank you so much!

YueSun0609 avatar Sep 12 '23 11:09 YueSun0609

Thank you for your kind release of the codes, I have successfully applied it to my own data and got good performace. But may I ask how to export a trained model to onnx format? In tools/export.py, you gave an example but missing "forward_test" function. I tried to write one, but when I run torch.onnx.export() function, it meets an error like "RuntimeError: Tracer cannot infer type of [{'boxes_3d': LiDARInstance3DBoxes", could you give me some clues on how to solve this issue? Thank you so much!

请问适配到自己的数据集时,训练需要采用和nuSence一样的数据格式吗?训练也需要输入六张图像和一个lidar的点云吗(如果只有一个front图像可以有效训练吗?)

evil-master avatar Oct 25 '23 06:10 evil-master

I got the same error, can anyone give some solution? thanks

ueke616 avatar Feb 05 '24 11:02 ueke616

We currently do not support exporting the model to ONNX, but it is under consideration for future updates.

zhijian-liu avatar May 04 '24 01:05 zhijian-liu