swjtulinxi / LoFTR_TRT

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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}
}

About

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

License:Apache License 2.0


Languages

Language:Python 100.0%