astoken commited on
Commit
a3395a2
·
verified ·
1 Parent(s): 2bfe492

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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 matching import get_matcher
18
 
19
  matcher = get_matcher("topicfm-plus", device="cuda")
20
- mkpts0, mkpts1, _, _, _, _ = matcher(img0, img1)
 
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)