reverted to torchvision transformations

model.py

@@ -21,10 +21,10 @@ SAMPLES_PATH = "./samples/*.jpg"
 
 class ImageLoader():
     def __init__(self):
-        self.transformer = torchvision.transforms.Compose([torchvision.transforms.ToTensor(),
-                               ImageTransformsFactory.create("global_resize::{'size':255}"),
-                               ImageTransformsFactory.create("center_crop::{'size':224}"),
-                               ImageTransformsFactory.create("normalize"),
+        self.transformer = torchvision.transforms.Compose([
+                               torchvision.transforms.Resize(256),
+                               torchvision.transforms.CenterCrop(224),
+                               torchvision.transforms.Normalize((.485, .456, .406), (.229, .224, .225)),
                                torchvision.transforms.ToTensor()])
     def load(self, im_path):
         im = torch.FloatTensor(self.transformer(Image.open(im_path))).unsqueeze(0)