/LoFTR_TRT

TensorRT adaptation of LoFTR: Detector-Free Local Feature Matching with Transformers

Primary LanguagePythonApache License 2.0Apache-2.0

Project development was moved to the https://github.com/Kolkir/Coarse_LoFTR_TRT

LoFTR_TRT

TensorRT adaptation of LoFTR: Detector-Free Local Feature Matching with Transformers

This a clone of the original LoFTR repository. The code was adapted for compatibility with TRTorch compile, especially dependencies to einsum and einops were removed.

The goal is to create TensorRT optimized inference PyTorch script that can be used on NVidia Jetson Nano device with reasonable performance.

For weights download please use original url that was provider by paper authors, the outdoor-ds file is only supported.

TODO:

  • [] Share the complete optimized PyTorch script

Paper:

@article{sun2021loftr,
  title={{LoFTR}: Detector-Free Local Feature Matching with Transformers},
  author={Sun, Jiaming and Shen, Zehong and Wang, Yuang and Bao, Hujun and Zhou, Xiaowei},
  journal={{CVPR}},
  year={2021}
}