temp

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 
 import torch
+from torchvision import transforms
 
 import fire_network
 
@@ -33,8 +34,8 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
     im1_tensor = transform(im1)
     im2_tensor = transform(im2)
 
-    im1_cv = cv2.imread(im1)
-    im2_cv = cv2.imread(im2)
+    # im1_cv = cv2.imread(im1)
+    # im2_cv = cv2.imread(im2)
 
     # extract features
     with torch.no_grad():
@@ -48,9 +49,11 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
         attns2 = output2[1]
         strenghts2 = output2[2]
 
-    print(feats1.shape)
-    print(attns1.shape)
-    print(strenghts1.shape)
+    print(len(feats1))
+    # print(feats1.shape)
+    print(feats1[0].shape)
+    # print(attns1.shape)
+    # print(strenghts1.shape)