pankaj-kaushik commited on
Commit
2c4515c
·
verified ·
1 Parent(s): bc6d22d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +34 -3
README.md CHANGED
@@ -1,8 +1,39 @@
1
  ---
2
  license: apache-2.0
3
  tags:
4
- - feature-matching
5
  - computer-vision
 
6
  - onnx
7
- - spatialhub
8
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
  tags:
 
4
  - computer-vision
5
+ - feature-matching
6
  - onnx
7
+ - efficient-loftr
8
+ pipeline_tag: image-feature-extraction
9
+ ---
10
+
11
+ # EfficientLoFTR ONNX Weights
12
+
13
+ This repository contains the ONNX-optimized weights for **EfficientLoFTR**, a model used for finding matching points between pairs of images.
14
+
15
+ By converting the original PyTorch model weights into the ONNX format, these files allow you to run fast feature-matching inference on both CPU and GPU without needing to install the heavy PyTorch framework.
16
+
17
+ ## Available Files
18
+ * **`eloftr_outdoor_full.onnx`**: The standard version of the model, optimized for the best matching quality.
19
+ * **`eloftr_outdoor_opt.onnx`**: An efficiency-focused version of the model, optimized for faster inference speed.
20
+
21
+ ---
22
+
23
+ ## How to Use
24
+
25
+ The easiest way to load and use these files is through the **[spatialhub](https://github.com/pankajkaushik12/spatialhub)** Python library.
26
+
27
+
28
+ ## Original Citation
29
+ If you use these models in academic work, please cite the original authors:
30
+ ```bibtex
31
+ @inproceedings{wang2022efficientloftr,
32
+ title={EfficientLoFTR: Semi-Dense Local Feature Matching with Sparse Transformers},
33
+ author={Wang, Yanzhao and Geng, Yuwei and Jiang, Zheng and Zhao, Yihong and Jin, Shisheng and Lin, Siyu and Han, Feng},
34
+ booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
35
+ year={2022}
36
+ }
37
+ ```
38
+
39
+ **Note**: This repository provides pre-converted weights for inference purposes.