thohemp / 6DRepNet

Official Pytorch implementation of 6DRepNet: 6D Rotation representation for unconstrained head pose estimation.
MIT License
550 stars 72 forks source link

Model convert #25

Closed June1124 closed 2 years ago

June1124 commented 2 years ago

I want to use libtorch to infer, it seems that the '.pt' format is a must for C++, how to convert '.pth' to '.pt'?

OuterSpaceTraveller commented 2 years ago

https://pytorch.org/tutorials/advanced/cpp_export.html