Spaces:
Build error
Build error
edits
Browse files
app.py
CHANGED
@@ -56,7 +56,7 @@ class ImgDataset(data.Dataset):
|
|
56 |
def __len__(self):
|
57 |
return len(self.images)
|
58 |
|
59 |
-
|
60 |
|
61 |
|
62 |
|
@@ -80,8 +80,8 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50, sf_ids='
|
|
80 |
loader = torch.utils.data.DataLoader(dataset_, shuffle=False, pin_memory=True)
|
81 |
|
82 |
|
83 |
-
|
84 |
-
|
85 |
|
86 |
im1_cv = np.array(im1)[:, :, ::-1].copy()
|
87 |
im2_cv = np.array(im2)[:, :, ::-1].copy()
|
|
|
56 |
def __len__(self):
|
57 |
return len(self.images)
|
58 |
|
59 |
+
# ---------------------------------------
|
60 |
|
61 |
|
62 |
|
|
|
80 |
loader = torch.utils.data.DataLoader(dataset_, shuffle=False, pin_memory=True)
|
81 |
|
82 |
|
83 |
+
im1_tensor = transform(im1).unsqueeze(0)
|
84 |
+
im2_tensor = transform(im2).unsqueeze(0)
|
85 |
|
86 |
im1_cv = np.array(im1)[:, :, ::-1].copy()
|
87 |
im2_cv = np.array(im2)[:, :, ::-1].copy()
|