Skip to content

Commit 8e379b0

Browse files
committed
update README
1 parent a17ed70 commit 8e379b0

File tree

7 files changed

+49
-6
lines changed

7 files changed

+49
-6
lines changed

README.md

Lines changed: 49 additions & 5 deletions
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). 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). Might consider supporting more if requested
2727
- [ ] Batch-inference
2828

2929
## Installation
@@ -164,8 +164,8 @@ the following result can be obtained
164164
---
165165

166166
<p align="center" width="100%">
167-
<img width="45%" align=top src="docs/images/dogs_maskrcnn_result.jpg">
168-
<img width="45%" align=top src="docs/images/indoor_maskrcnn_result.jpg">
167+
<img width="45%" height="250" src="docs/images/dogs_maskrcnn_result.jpg">
168+
<img width="45%" height="250" src="docs/images/indoor_maskrcnn_result.jpg">
169169
</p>
170170

171171
<details>
@@ -271,8 +271,8 @@ wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yo
271271
</p>
272272

273273
<p align="center" width="100%">
274-
<img width="45%" align=top src="docs/images/sample_city_scapes_result.jpg">
275-
<img width="45%" align=top src="docs/images/odaiba_result.jpg">
274+
<img width="45%" height="250" align=top src="docs/images/sample_city_scapes_result.jpg">
275+
<img width="45%" height="250" align=top src="docs/images/odaiba_result.jpg">
276276
</p>
277277

278278
<details>
@@ -335,3 +335,47 @@ wget https://raw.githubusercontent.com/StaRainJ/Multi-modality-image-matching-da
335335
</details>
336336

337337
<p align="right">(<a href="#readme-top">back to top</a>)</p>
338+
339+
### [SuperGlue](https://arxiv.org/pdf/1911.11763.pdf)
340+
341+
---
342+
343+
<p align="center" width="100%">
344+
<img width="49%" height="250" src="docs/images/ComputerVision_VisionCS_0.jpg">
345+
<img width="49%" height="250" src="docs/images/ComputerVision_VN_30.jpg">
346+
</p>
347+
348+
<p align="center" width="100%">
349+
<img width="49%" height="250" src="docs/images/RemoteSensing_CS1.jpg">
350+
<img width="49%" height="250" src="docs/images/RemoteSensing_CS5.jpg">
351+
</p>
352+
353+
<details>
354+
<summary>Usage</summary>
355+
356+
- Convert SuperPoint's pretrained weights to onnx format: Follow the above instruction
357+
358+
- Convert SuperGlue's pretrained weights to onnx format
359+
360+
```bash
361+
git submodule update --init --recursive
362+
python3 -m pip install -r scripts/superglue/requirements.txt
363+
python3 -m pip install -r scripts/superglue/SuperGluePretrainedNetwork/requirements.txt
364+
python3 scripts/superglue/convert_to_onnx.py
365+
```
366+
367+
- 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
368+
369+
- Test inference apps
370+
371+
```bash
372+
./build/examples/super_glue /path/to/super_point.onnx /path/to/super_glue.onnx /path/to/1st/image /path/to/2nd/image
373+
```
374+
375+
- Note: the pretrained superglue model tends to fail on Day-Night image pairs (which also happens for SuperGlue + KNN Match + Lowe's Ratio Test case).
376+
377+
<p align="center" width="100%">
378+
<img width="49%" height="250" src="docs/images/ComputerVision_VisionDN_1.jpg">
379+
</p>
380+
381+
</details>
147 KB
Loading
124 KB
Loading
109 KB
Loading

docs/images/RemoteSensing_CS1.jpg

102 KB
Loading

docs/images/RemoteSensing_CS5.jpg

239 KB
Loading

scripts/superglue/README.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@ python3 -m pip install -r SuperGluePretrainedNetwork/requirements.txt
2121

2222
---
2323

24-
2524
- export onnx weights
2625

2726
```

0 commit comments

Comments
 (0)