search

herohuyongtao / deeptag-pytorch

Official implementation of paper [DeepTag: A General Framework for Fiducial Marker Design and Detection]
https://herohuyongtao.github.io/research/publications/deep-tag/
101 stars 19 forks source link

how can i convert pth to onnx model #5

Closed jcyhcs closed 2 years ago

jcyhcs commented 2 years ago

hi,professor: how can convert pth model to onnx model? please help!

sskzzm commented 2 years ago

As the network implementation and the model weights are both provided, please follow the PyTorch official tutorial (https://pytorch.org/docs/stable/onnx.html) to convert our model to ONNX model.