Upload processor

processing_encoder.py

@@ -5,6 +5,8 @@ from transformers.image_transforms import resize, center_crop, normalize
 from transformers.utils.generic import TensorType
 from transformers.image_processing_utils import BatchFeature
 from PIL import Image
+import torchvision.transforms 
+import numpy as np
 
 
 class EncoderImageProcessor(BaseImageProcessor):
@@ -44,17 +46,18 @@ class EncoderImageProcessor(BaseImageProcessor):
         Returns:
             torch.Tensor: Transformed image tensor.
         """
+        image = np.array(image)
         # Resize
-        image = resize(image, size=self.input_size, default_to_square=False)
+        image = resize(image, size=self.input_size)
 
         # Center Crop
         image = center_crop(image, size=self.input_size)
-
+        image = normalize(image, mean=self.mean, std=self.std)
         # Convert to tensor and normalize
-        image_tensor = torch.tensor(image, dtype=torch.float32).permute(2, 0, 1) / 255.0  # Convert to CHW format
-        image_tensor = normalize(image_tensor, mean=self.mean, std=self.std)
+        image = torch.Tensor(image).to(torch.float32).permute(2,0,1) / 255.0  # Convert to CHW format
+       
 
-        return image_tensor
+        return image
 
     def preprocess(
         self,