Spaces:
Build error
Build error
temp
Browse files
app.py
CHANGED
|
@@ -47,8 +47,8 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
|
|
| 47 |
im1_tensor = transform(im1).unsqueeze(0)
|
| 48 |
im2_tensor = transform(im2).unsqueeze(0)
|
| 49 |
|
| 50 |
-
im1_cv =
|
| 51 |
-
im2_cv =
|
| 52 |
|
| 53 |
# extract features
|
| 54 |
with torch.no_grad():
|
|
|
|
| 47 |
im1_tensor = transform(im1).unsqueeze(0)
|
| 48 |
im2_tensor = transform(im2).unsqueeze(0)
|
| 49 |
|
| 50 |
+
im1_cv = np.array(im1)[:, :, ::-1].copy()
|
| 51 |
+
im2_cv = np.array(im2)[:, :, ::-1].copy()
|
| 52 |
|
| 53 |
# extract features
|
| 54 |
with torch.no_grad():
|