YannisK commited on
Commit
3ea46fa
1 Parent(s): a809352
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -193,8 +193,9 @@ def generate_matching_superfeatures(im1, im2, Imagenet_model=False, scale_id=6,
193
  # att = resize(att, im1.size)
194
  mask2d = zip(*np.where(att==255))
195
  for m,n in mask2d:
196
- col_ = col.colors[j] if j < 7 else col.colors[j+1]
197
- if j == 0: col_ = col.colors[9]
 
198
  col_ = 255*np.array(colors.to_rgba(col_))[:3]
199
  img1rsz[m,n, :] = col_[::-1]
200
 
 
193
  # att = resize(att, im1.size)
194
  mask2d = zip(*np.where(att==255))
195
  for m,n in mask2d:
196
+ col_ = col.colors[j]
197
+ # col_ = col.colors[j] if j < 7 else col.colors[j+1]
198
+ # if j == 0: col_ = col.colors[9]
199
  col_ = 255*np.array(colors.to_rgba(col_))[:3]
200
  img1rsz[m,n, :] = col_[::-1]
201