temp

app.py

@@ -34,8 +34,8 @@ col = plt.get_cmap('tab10')
 
 def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
     
-    im1_tensor = transform(im1)
-    im2_tensor = transform(im2)
+    im1_tensor = transform(im1).unsqueeze(0)
+    im2_tensor = transform(im2).unsqueeze(0)
 
     # im1_cv = cv2.imread(im1)
     # im2_cv = cv2.imread(im2)
@@ -55,6 +55,7 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
     print(len(feats1))
     # print(feats1.shape)
     print(feats1[0].shape)
+    print(attns1[0].shape)
     # print(attns1.shape)
     # print(strenghts1.shape)