Update README.md
Browse files
README.md
CHANGED
|
@@ -14,13 +14,13 @@ Plus variant of TopicFM with 8-head attention and linear attention mechanism.
|
|
| 14 |
## Usage
|
| 15 |
|
| 16 |
```python
|
| 17 |
-
from
|
| 18 |
|
| 19 |
matcher = get_matcher("topicfm-plus", device="cuda")
|
| 20 |
-
|
|
|
|
| 21 |
```
|
| 22 |
|
| 23 |
## Links
|
| 24 |
-
- [TopicFM](https://huggingface.co/image-matching-models/topicfm)
|
| 25 |
- [Original repository](https://github.com/TruongKhang/TopicFM)
|
| 26 |
- [Image Matching Models](https://github.com/gmberton/image-matching-models)
|
|
|
|
| 14 |
## Usage
|
| 15 |
|
| 16 |
```python
|
| 17 |
+
from imm import get_matcher
|
| 18 |
|
| 19 |
matcher = get_matcher("topicfm-plus", device="cuda")
|
| 20 |
+
result = matcher(img0, img1)
|
| 21 |
+
# result.keys() = # result.keys() = ['num_inliers', 'H', 'all_kpts0', 'all_kpts1', 'all_desc0', 'all_desc1', 'matched_kpts0', 'matched_kpts1', 'inlier_kpts0', 'inlier_kpts1']
|
| 22 |
```
|
| 23 |
|
| 24 |
## Links
|
|
|
|
| 25 |
- [Original repository](https://github.com/TruongKhang/TopicFM)
|
| 26 |
- [Image Matching Models](https://github.com/gmberton/image-matching-models)
|