edits

app.py

@@ -37,24 +37,24 @@ transform = transforms.Compose([
         ])
 # ---------------------------------------
 
-class ImgDataset(data.Dataset):
-    
-    def __init__(self, images, imsize):
-        self.images = images
-        self.imsize = imsize
-        self.transform = transforms.Compose([transforms.ToTensor(), \
-            transforms.Normalize(**dict(zip(["mean", "std"], net.runtime['mean_std'])))])
+# class ImgDataset(data.Dataset):
+
+#     def __init__(self, images, imsize):
+#         self.images = images
+#         self.imsize = imsize
+#         self.transform = transforms.Compose([transforms.ToTensor(), \
+#             transforms.Normalize(**dict(zip(["mean", "std"], net.runtime['mean_std'])))])
 
 
-    def __getitem__(self, index):
-        img = self.images[index]
-        img.thumbnail((self.imsize, self.imsize), Image.Resampling.LANCZOS)
-        print('after imresize:', img.size)
-        return  self.transform(img)
+#     def __getitem__(self, index):
+#         img = self.images[index]
+#         img.thumbnail((self.imsize, self.imsize), Image.Resampling.LANCZOS)
+#         print('after imresize:', img.size)
+#         return  self.transform(img)
 
 
-    def __len__(self):
-        return len(self.images)
+#     def __len__(self):
+#         return len(self.images)
 
 # ---------------------------------------    
 
@@ -132,7 +132,7 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50, sf_ids='
     print('im1:', im1.size)
     print('img1rsz:', img1rsz.shape)
     for j, att in enumerate(all_att_bin1):
-        att = cv2.resize(att, im1.size, interpolation=cv2.INTER_NEAREST)
+        att = cv2.resize(att, img1rsz.shape[:2], 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(shape)
@@ -149,7 +149,7 @@ def generate_matching_superfeatures(im1, im2, scale_id=6, threshold=50, sf_ids='
     print('im2:', im2.size)
     print('img2rsz:', img2rsz.shape)
     for j, att in enumerate(all_att_bin2):
-        att = cv2.resize(att, im2.size, interpolation=cv2.INTER_NEAREST)
+        att = cv2.resize(att, img2rsz.shape[:2], 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.shape)