YannisK commited on
Commit
cb3d625
1 Parent(s): f7b2f96
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -149,12 +149,12 @@ def generate_matching_superfeatures(im1, im2, Imagenet_model=False, scale_id=6,
149
  sf_idx_ = map(int, sf_ids.strip().split(','))
150
  n_sf_ids = len(sf_idx_)
151
 
152
- if only_matching:
153
- if random_mode:
154
- sf_idx_ = [int(jj) for jj in ind_match[np.random.randint(len(list(ind_match)), size=n_sf_ids)].numpy()]
155
- sf_idx_ = list( dict.fromkeys(sf_idx_) )
156
- else:
157
- sf_idx_ = [i for i in sf_idx_ if i in list(ind_match)]
158
 
159
 
160
 
 
149
  sf_idx_ = map(int, sf_ids.strip().split(','))
150
  n_sf_ids = len(sf_idx_)
151
 
152
+ # if only_matching:
153
+ if random_mode:
154
+ sf_idx_ = [int(jj) for jj in ind_match[np.random.randint(len(list(ind_match)), size=n_sf_ids)].numpy()]
155
+ sf_idx_ = list( dict.fromkeys(sf_idx_) )
156
+ else:
157
+ sf_idx_ = [i for i in sf_idx_ if i in list(ind_match)]
158
 
159
 
160