YannisK commited on
Commit
388a978
1 Parent(s): 5d89783
Files changed (1) hide show
  1. app.py +2 -2
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)