Skip to content

Commit b24c203

Browse files
committed
update loftr README
1 parent 861b44f commit b24c203

File tree

3 files changed

+37
-3
lines changed

3 files changed

+37
-3
lines changed

README.md

Lines changed: 34 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ Hope that they both are helpful for your work.
2323
## TODO
2424

2525
- [x] Support inference of multi-inputs, multi-outputs
26-
- [x] Examples for famous models, like yolov3, mask-rcnn, [ultra-light-weight face detector](https://github.com/Linzaer/Ultra-Light-Fast-Generic-Face-Detector-1MB), [yolox](https://github.com/Megvii-BaseDetection/YOLOX), [PaddleSeg](https://github.com/PaddlePaddle/PaddleSeg/tree/release/2.3), [SuperPoint](https://github.com/magicleap/SuperPointPretrainedNetwork), [SuperGlue](https://github.com/magicleap/SuperGluePretrainedNetwork/tree/ddcf11f42e7e0732a0c4607648f9448ea8d73590). Might consider supporting more if requested.
26+
- [x] Examples for famous models, like yolov3, mask-rcnn, [ultra-light-weight face detector](https://github.com/Linzaer/Ultra-Light-Fast-Generic-Face-Detector-1MB), [yolox](https://github.com/Megvii-BaseDetection/YOLOX), [PaddleSeg](https://github.com/PaddlePaddle/PaddleSeg/tree/release/2.3), [SuperPoint](https://github.com/magicleap/SuperPointPretrainedNetwork), [SuperGlue](https://github.com/magicleap/SuperGluePretrainedNetwork/tree/ddcf11f42e7e0732a0c4607648f9448ea8d73590), [LoFTR](https://zju3dv.github.io/loftr/). Might consider supporting more if requested.
2727
- [x] (Minimal^^) Support for TensorRT backend
2828
- [ ] Batch-inference
2929

@@ -385,3 +385,36 @@ python3 scripts/superglue/convert_to_onnx.py
385385
</details>
386386

387387
<p align="right">(<a href="#readme-top">back to top</a>)</p>
388+
389+
### [LoFTR](https://zju3dv.github.io/loftr/)
390+
391+
---
392+
393+
<p align="center" width="100%">
394+
<img width="100%" src="docs/images/loftr.jpg">
395+
</p>
396+
397+
<details>
398+
<summary>Usage</summary>
399+
400+
- Download [LoFTR](https://github.com/zju3dv/LoFTR) weights indoor*ds_new.ckpt from [HERE](https://drive.google.com/drive/folders/1xu2Pq6mZT5hmFgiYMBT9Zt8h1yO-3SIp). (LoFTR's [latest commit](b4ee7eb0359d0062e794c99f73e27639d7c7ac9f) seems to be only compatible with the new weights (Ref: https://github.com/zju3dv/LoFTR/issues/48). Hence, this onnx cpp application is only compatible with \_indoor_ds_new.ckpt* weights)
401+
402+
- Convert LoFTR's pretrained weights to onnx format
403+
404+
```bash
405+
git submodule update --init --recursive
406+
python3 -m pip install -r scripts/loftr/requirements.txt
407+
python3 scripts/loftr/convert_to_onnx.py --model_path /path/to/indoor_ds_new.ckpt
408+
```
409+
410+
- Download test images from [this dataset](https://github.com/StaRainJ/Multi-modality-image-matching-database-metrics-methods): Or prepare some pairs of your own images
411+
412+
- Test inference apps
413+
414+
```bash
415+
./build/examples/loftr /path/to/loftr.onnx /path/to/loftr.onnx /path/to/1st/image /path/to/2nd/image
416+
```
417+
418+
</details>
419+
420+
<p align="right">(<a href="#readme-top">back to top</a>)</p>

docs/images/loftr.jpg

618 KB
Loading

scripts/loftr/README.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ python3 -m pip install -r requirements.txt
2727
```
2828
python3 convert_to_onnx.py --model_path /path/to/indoor_ds_new.ckpt
2929
```
30-
## Note ##
3130

32-
- The LoFTR's [latest commit](b4ee7eb0359d0062e794c99f73e27639d7c7ac9f) seems to be only compatible with the new weights (Ref: https://github.com/zju3dv/LoFTR/issues/48). Hence, this onnx cpp application is only compatible with *indoor_ds_new.ckpt* weights.
31+
## Note
32+
33+
- The LoFTR's [latest commit](b4ee7eb0359d0062e794c99f73e27639d7c7ac9f) seems to be only compatible with the new weights (Ref: https://github.com/zju3dv/LoFTR/issues/48). Hence, this onnx cpp application is only compatible with _indoor_ds_new.ckpt_ weights.

0 commit comments

Comments
 (0)