How to convert *.pth file to *.onnx?

45 次查看（过去 30 天）
Abbas
Abbas 2024-6-14
回答： 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

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

回答（1 个）

Sanjana
Sanjana 2024-6-15
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!

类别

在 Help Center 和 File Exchange 中查找有关 Electrical Block Libraries 的更多信息

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by