YannisK commited on
Commit
b489890
1 Parent(s): 092e211
Files changed (1) hide show
  1. app.py +3 -3
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
- # 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()
 
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()