YannisK commited on
Commit
082f0c5
·
1 Parent(s): e7c9542
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -86,7 +86,7 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
86
  # att = cv2.resize(att, imgz[i].shape[:2][::-1], interpolation=cv2.INTER_NEAREST)
87
  # att = cv2.resize(att, imgz[i].shape[:2][::-1], interpolation=cv2.INTER_CUBIC)
88
  # att = cv2.resize(att, imgz[i].shape[:2][::-1])
89
- att = att.resize(im1.size)
90
  mask2d = zip(*np.where(att==255))
91
  for m,n in mask2d:
92
  col_ = col.colors[j] if j < 7 else col.colors[j+1]
@@ -101,8 +101,8 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
101
  # att = cv2.resize(att, imgz[i].shape[:2][::-1], interpolation=cv2.INTER_NEAREST)
102
  # att = cv2.resize(att, imgz[i].shape[:2][::-1], interpolation=cv2.INTER_CUBIC)
103
  # att = cv2.resize(att, imgz[i].shape[:2][::-1])
104
- att = att.resize(im2.size)
105
- print('att:', att.size)
106
  mask2d = zip(*np.where(att==255))
107
  for m,n in mask2d:
108
  col_ = col.colors[j] if j < 7 else col.colors[j+1]
 
86
  # att = cv2.resize(att, imgz[i].shape[:2][::-1], interpolation=cv2.INTER_NEAREST)
87
  # att = cv2.resize(att, imgz[i].shape[:2][::-1], interpolation=cv2.INTER_CUBIC)
88
  # att = cv2.resize(att, imgz[i].shape[:2][::-1])
89
+ att = att.resize(im1.shape)
90
  mask2d = zip(*np.where(att==255))
91
  for m,n in mask2d:
92
  col_ = col.colors[j] if j < 7 else col.colors[j+1]
 
101
  # att = cv2.resize(att, imgz[i].shape[:2][::-1], interpolation=cv2.INTER_NEAREST)
102
  # att = cv2.resize(att, imgz[i].shape[:2][::-1], interpolation=cv2.INTER_CUBIC)
103
  # att = cv2.resize(att, imgz[i].shape[:2][::-1])
104
+ att = att.resize(im2.shape)
105
+ print('att:', att.shape)
106
  mask2d = zip(*np.where(att==255))
107
  for m,n in mask2d:
108
  col_ = col.colors[j] if j < 7 else col.colors[j+1]