How to convert *.pth file to *.onnx?
45 views (last 30 days)
Show older comments
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 Comments
Answers (1)
Sanjana
on 15 Jun 2024
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:
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:
Hope this helps!
0 Comments
See Also
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!