Spaces:
Build error
Build error
edits
Browse files
app.py
CHANGED
|
@@ -76,8 +76,8 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50, sf_ids='
|
|
| 76 |
sf_idx_ = map(int, sf_ids.strip().split(','))
|
| 77 |
|
| 78 |
|
| 79 |
-
dataset_ = ImgDataset(images=[im1, im2], imsize=1024)
|
| 80 |
-
loader = torch.utils.data.DataLoader(dataset_, shuffle=False, pin_memory=True)
|
| 81 |
|
| 82 |
|
| 83 |
im1_tensor = transform(im1).unsqueeze(0)
|
|
|
|
| 76 |
sf_idx_ = map(int, sf_ids.strip().split(','))
|
| 77 |
|
| 78 |
|
| 79 |
+
# dataset_ = ImgDataset(images=[im1, im2], imsize=1024)
|
| 80 |
+
# loader = torch.utils.data.DataLoader(dataset_, shuffle=False, pin_memory=True)
|
| 81 |
|
| 82 |
|
| 83 |
im1_tensor = transform(im1).unsqueeze(0)
|