paulprt
/

swin-fish-classification

swin_classifier

swin-transformer

fish-classification

Model card Files Files and versions

Paul commited on Jan 15

Commit

3da5180

·

verified ·

1 Parent(s): 336cc28

Upload processor

Files changed (1) hide show

image_processing_swin.py +1 -1

image_processing_swin.py CHANGED Viewed

@@ -30,7 +30,7 @@ class MyCustomSwinProcessor(ViTImageProcessor):
         bottom = pad_h - top
         # Pad with white because its the dataset default background color
-        return transforms.functional.pad(image, (left, top, right, bottom), fill=255)
     def preprocess(self, images, **kwargs):
         images = [self.resize_and_pad(image, target_size=224) for image in images]

         bottom = pad_h - top
         # Pad with white because its the dataset default background color
+        return transforms.functional.pad(image, (left, top, right, bottom), fill=255).totensor()
     def preprocess(self, images, **kwargs):
         images = [self.resize_and_pad(image, target_size=224) for image in images]