show resize code update

model.py

@@ -26,13 +26,7 @@ class ImageLoader():
                                torchvision.transforms.Resize(256),
                                torchvision.transforms.CenterCrop(224),
                                torchvision.transforms.Normalize((.485, .456, .406), (.229, .224, .225))])
-        self.show_manip = torchvision.transforms.Compose(
-            [
-                torchvision.transforms.ToTensor(),
-                torchvision.transforms.Resize(500, max_size=500),
-                torchvision.transforms.ToPILImage()
-            ]
-        )
+        self.show_size=500
         
     def load(self, im_path):
         im = torch.FloatTensor(self.transformer(Image.open(im_path))).unsqueeze(0)
@@ -51,7 +45,12 @@ class ImageLoader():
         return text.lower()
     
     def show_resize(self, image):
-        return self.show_manip(image)
+        # ugh we need to do this manually cuz this is pytorch==0.8 not 1.9 lol
+        image = torchvision.transforms.functional.to_tensor(image)
+        x,y = image.shape[-2:]
+        ratio = float(self.show_size/max((x,y)))
+        image = torchvision.transforms.functional.resize(image, [int(x * ratio), int(y * ratio)])
+        return torch.transforms.functional.to_pil_image(image)
     
 
 class VirTexModel():