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