Skip to content

v0.1.0: LightGlue-ONNX
Compare
Choose a tag to compare
@fabio-sim fabio-sim released this 04 Jul 11:31
· 70 commits to main since this release
v0.1.0
1da3ab3
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

LightGlue-ONNX Models

This release provides exported ONNX models of the DISK, SuperPoint (feature extractors), and LightGlue (keypoint matcher) models. Both standalone models and end-to-end pipelines (*_end2end.onnx) are provided.

Individual Models

  • disk.onnx: DISK feature extractor.
  • disk_{N}.onnx: DISK feature extractor with max_num_keypoints=N.
  • disk_lightglue.onnx: LightGlue model trained on DISK features.
  • superpoint.onnx: SuperPoint feature extractor.
  • superpoint_{N}.onnx: SuperPoint feature extractor with max_num_keypoints=N.
  • superpoint_lightglue.onnx: LightGlue model trained on SuperPoint features.

End-to-end Pipelines

  • disk_lightglue_end2end.onnx: LightGlue model fused to DISK.
  • disk_{N}_lightglue_end2end.onnx: LightGlue model fused to DISK with max_num_keypoints=N.
  • superpoint_lightglue_end2end.onnx: LightGlue model fused to SuperPoint.
  • superpoint_{N}_lightglue_end2end.onnx: LightGlue model fused to SuperPoint with max_num_keypoints=N.

All models are dynamic with respect to image size. Note that a model exported with static input shapes may perform faster due to ONNX optimisations.

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

Assets 21
Loading