YannisK commited on
Commit
a809352
1 Parent(s): a92520a
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -154,7 +154,6 @@ def generate_matching_superfeatures(im1, im2, Imagenet_model=False, scale_id=6,
154
  sf_idx_ = np.random.randint(256, size=n_sf_ids)
155
  else:
156
  sf_idx_ = map(int, sf_ids.strip().split(','))
157
- n_sf_ids = len(sf_idx_)
158
 
159
  # if only_matching:
160
  if random_mode:
@@ -163,6 +162,8 @@ def generate_matching_superfeatures(im1, im2, Imagenet_model=False, scale_id=6,
163
  else:
164
  sf_idx_ = [i for i in sf_idx_ if i in list(ind_match)]
165
 
 
 
166
  # Store all binary SF att maps to show them all at once in the end
167
  all_att_bin1 = []
168
  all_att_bin2 = []
 
154
  sf_idx_ = np.random.randint(256, size=n_sf_ids)
155
  else:
156
  sf_idx_ = map(int, sf_ids.strip().split(','))
 
157
 
158
  # if only_matching:
159
  if random_mode:
 
162
  else:
163
  sf_idx_ = [i for i in sf_idx_ if i in list(ind_match)]
164
 
165
+ n_sf_ids = len(sf_idx_)
166
+
167
  # Store all binary SF att maps to show them all at once in the end
168
  all_att_bin1 = []
169
  all_att_bin2 = []