Export to ONNX

Question

Export to ONNX

williamhoole opened this issue 2 months ago · comments

Willows commented 2 months ago

Hi,

Is it possible to export the model to ONNX ?

Yifan Wang · Answer 1 · Tue Apr 23 2024 23:55:32 GMT+0800 (China Standard Time)

Hi,

I believe it is possible to export the model to ONNX since our model doesn't involve any necessary dynamic control flow, and all the PyTorch functions we used are common. However, we currently do not have plans to offer an ONNX version because we lack experience with ONNX optimization. We would very much welcome any attempts to try this out.

Willows · Answer 2 · Thu Apr 25 2024 17:24:58 GMT+0800 (China Standard Time)

Hi thank you very much for your reply! I have found a repository that converts the original LOFTR to ONNX. This should also work for EfficientLoFTR. I will try adjust the code for this repository. Here is the link: loftr2onnx

Tom · Answer 3 · Fri May 31 2024 23:10:30 GMT+0800 (China Standard Time)

@williamhoole did you manage to adapt the exporter to eloftr? I managed to get an .onnx after some minor tweaks in the forward, but the onnx inference fails bc of size mismatches. Thanks!