use image transforms

model.py

@@ -21,8 +21,8 @@ SAMPLES_PATH = "./samples/*.jpg"
 
 class ImageLoader():
     def __init__(self):
-        self.transformer = torchvision.transforms.Compose([ImageTransformsFactory.create("smallest_resize"),
-                               ImageTransformsFactory.create("center_crop"),
+        self.transformer = torchvision.transforms.Compose([ImageTransformsFactory.create("global_resize::{'size':255}"),
+                               ImageTransformsFactory.create("center_crop::{'size':224}"),
                                ImageTransformsFactory.create("normalize"),
                                torchvision.transforms.ToTensor()])
     def load(self, im_path):