fix eval

mACHINE-LEARNINGS/eval.py

@@ -1,11 +1,11 @@
 import warnings; warnings.filterwarnings('ignore')
 import torch
 from torch.utils.data import DataLoader
-from torchvision.transforms import Resize
 from torchvision.io import read_image, ImageReadMode
 from classes.MemeDataset import MemeDataset
 from models.DankCNN import DankCNN
 import argparse
+from torchvision import transforms
 
 cls = {0: 'dank', 1: 'normie'}
 
@@ -23,9 +23,15 @@ def eval(image):
         - Load all data: train, test, validation
     """
     image = read_image(image, ImageReadMode.RGB).float()
-    resize = Resize(size = (160, 160))
-    image = resize(image)
-
+    data_transforms = {
+    'test': transforms.Compose([
+        transforms.ToPILImage(),
+        transforms.Resize(320),
+        transforms.ToTensor(),
+        transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
+    ])
+    }
+    image = data_transforms['test'](image)
     """
     MODEL INITIALIZATION
         - optimizer: Adam with weight decay as regularization technique