search

fabio-sim / LightGlue-ONNX

ONNX-compatible LightGlue: Local Feature Matching at Light Speed. Supports TensorRT, OpenVINO
Apache License 2.0
375 stars 34 forks source link

".trt.onnx"export example #70

Closed WYKXD closed 5 months ago

WYKXD commented 8 months ago

Hello, "superpoint_lightglue.trt.onnx"

Is there an export example of the file? I see examples of "_fp16.onnx" and "_fused.onnx" in the code, but I don't see examples of trt.onnx

fabio-sim commented 7 months ago

Hi @WYKXD, thank you for your interest in LightGlue-ONNX.

The .trt.onnx refers to the ONNX model that has undergone shape inference and can therefore be consumed by TensorRT. You can use the following command:

python -m onnxruntime.tools.symbolic_shape_infer --auto_merge --input superpoint_lightglue.onnx --output superpoint_lightglue.trt.onnx

Hope this helps :)