WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.99k stars 521 forks source link

Convert .pt to Onnx #180

Open MasIgor opened 2 years ago

MasIgor commented 2 years ago

Hello, i am very new in this sector. I trained my network in collab and have a .pt file now. iwould liek to import this into my .net project using ML.net but it requires a OnnX file. I was not able so far to find a way to convert a YoloR weight file to a onnx.

Could you pleae help me out on this? thank you

otsebriy commented 2 years ago

Onnx export

MasIgor commented 2 years ago

Thank you very much!