YannisK commited on
Commit
f981819
1 Parent(s): 74bae21
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -47,8 +47,8 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
47
  im1_tensor = transform(im1).unsqueeze(0)
48
  im2_tensor = transform(im2).unsqueeze(0)
49
 
50
- im1_cv = cv2.imread(im1)
51
- im2_cv = cv2.imread(im2)
52
 
53
  # extract features
54
  with torch.no_grad():
 
47
  im1_tensor = transform(im1).unsqueeze(0)
48
  im2_tensor = transform(im2).unsqueeze(0)
49
 
50
+ im1_cv = np.array(im1)[:, :, ::-1].copy()
51
+ im2_cv = np.array(im2)[:, :, ::-1].copy()
52
 
53
  # extract features
54
  with torch.no_grad():