solve error shapes

help_function.py

@@ -2,6 +2,7 @@ import torch
 import open_clip
 from torchvision import transforms
 from torchvision.transforms import ToPILImage
+import torch.nn.functional as F
 
 class help_function:
     def __init__(self):
@@ -24,6 +25,9 @@ class help_function:
 
     def get_image_inversion(self, image):
         image = self.transform(image)
+        if not image.shape == torch.Size([3, 224, 224]):
+            image = image.reshape(1,3,image.shape[1],image.shape[2])
+            image = F.interpolate(image, [224,224], mode='bilinear', align_corners=True)
         w_inversion = self.encoder(image.reshape(1,3,224,224)).reshape(1,16,512)
         return w_inversion + self.mean_person