l45k
/

lenet

Image Classification

Model card Files Files and versions

l45k commited on 23 days ago

Commit

2ebaa9f

·

verified ·

1 Parent(s): 1f2e2c7

Upload processor

Files changed (1) hide show

preprocessor_lenet.py +1 -1

preprocessor_lenet.py CHANGED Viewed

@@ -28,7 +28,7 @@ class LeNetProcessor(BaseImageProcessor):
             images = [images]
         transform = v2.Compose([
-            v2.RandomResizedCrop(size=(28, 28), antialias=True),
             v2.ToDtype(torch.float32, scale=True),
             v2.Normalize(
                 mean=[0.1307],

             images = [images]
         transform = v2.Compose([
+            v2.Resize(size=(28, 28), antialias=True),
             v2.ToDtype(torch.float32, scale=True),
             v2.Normalize(
                 mean=[0.1307],