Update app.py
Browse files
app.py
CHANGED
|
@@ -109,7 +109,7 @@ def image_matching_no_pyramid(query_img, target_img, visualize=True):
|
|
| 109 |
mkpts0 = kpts0[valid]
|
| 110 |
mkpts1 = kpts1[matches[valid]]
|
| 111 |
mconf = conf[valid]
|
| 112 |
-
color = cm.jet(mconf.cpu())[:len(mkpts0)]
|
| 113 |
|
| 114 |
valid_count = np.sum(valid.tolist())
|
| 115 |
|
|
|
|
| 109 |
mkpts0 = kpts0[valid]
|
| 110 |
mkpts1 = kpts1[matches[valid]]
|
| 111 |
mconf = conf[valid]
|
| 112 |
+
color = cm.jet(mconf.detach().cpu().numpy())[:len(mkpts0)]
|
| 113 |
|
| 114 |
valid_count = np.sum(valid.tolist())
|
| 115 |
|