Update README.md
Browse files
README.md
CHANGED
|
@@ -1,8 +1,39 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
tags:
|
| 4 |
-
- feature-matching
|
| 5 |
- computer-vision
|
|
|
|
| 6 |
- onnx
|
| 7 |
-
-
|
| 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.
|