temp

app.py

@@ -63,8 +63,8 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
     print(strenghts1.shape, strenghts2.shape)
     
     # Store all binary SF att maps to show them all at once in the end
-    all_att_bin1 = defaultdict(list)
-    all_att_bin2 = defaultdict(list)
+    all_att_bin1 = []
+    all_att_bin2 = []
     for n, i in enumerate(sf_idx_):
         # all_atts[n].append(attn[j][scale_id][0,i,:,:].numpy())
         att_heat = np.array(attns1[0,i,:,:].numpy(), dtype=np.float32)
@@ -94,11 +94,13 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50):
     fin_img.append(img1rsz)
             
     img2rsz = np.copy(im2)
+    print(img2rsz.size)
     for j, att in enumerate(all_att_bin2):
         # att = cv2.resize(att, imgz[i].shape[:2][::-1], interpolation=cv2.INTER_NEAREST)
         # att = cv2.resize(att, imgz[i].shape[:2][::-1], interpolation=cv2.INTER_CUBIC)
         # att = cv2.resize(att, imgz[i].shape[:2][::-1])
         att = att.resize(im2.size)
+        print('att:', att.size)
         mask2d = zip(*np.where(att==255))
         for m,n in mask2d:
             col_ = col.colors[j] if j < 7 else col.colors[j+1]