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