How to convert .pth file to .onnx?

Question

Abbas 2024-6-14

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/2128691-how-to-convert-pth-file-to-onnx

回答： Sanjana 2024-6-15

Hi,

I used Pointpillars for 3D object detection on my own dataset (just using *.bin files and their *.txt labels) using Pytorch (Cuda). The format of the trained model is *.pth. I would like to convert the *.pth file to *.onnx format. I think I should define a model (Pointpillars) and a dummy tensor/input in this regard. I am not sure how to define them. Is there any MATLAB example/library in this regard?

I appreciate any help in advance.

Thank you,

Abbas

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Sanjana 2024-6-15

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/2128691-how-to-convert-pth-file-to-onnx#answer_1472231

Hello,

Currently, MATLAB does not have built-in functionality to directly convert models saved in the ".pth" format to the ".onnx" format.

For guidance on converting ".pth" to ".onnx," please refer to the following link:

https://github.com/danielgatis/rembg/issues/193#issuecomment-1049586873

Once the model has been saved in the ".onnx" format, you can utilize the "importNetworkFromONNX" functionality in MATLAB to load the pre-trained network.

For more information, visit:

https://www.mathworks.com/help/deeplearning/ref/importnetworkfromonnx.html

Hope this helps!