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():
|