Skip to content

v1.0.0: Fused LightGlue-ONNX

Latest
Compare
Choose a tag to compare
@fabio-sim fabio-sim released this 03 Oct 17:54
· 12 commits to main since this release
3140781

Fused LightGlue ONNX Models

This release provides optimized LightGlue ONNX models that have been fused (multi-head attention fusion). MultiHeadAttention is a contrib operator which is intended to be run using ONNXRuntime.

All models are dynamic and have undergone symbolic shape inference, but for TensorRT via ORT, it is recommended to use superpoint_lightglue_fused_fp16.onnx. For pure TensorRT, please use superpoint_lightglue.trt.onnx as the source model for building the engine.

For reproducibility, the environment in which these models were exported is also provided (pip-freeze.txt). All models were exported with ONNX opset version 17.