zamborg commited on
Commit
9fb86d0
·
1 Parent(s): 2ca42e8

syntax fix

Browse files
Files changed (1) hide show
  1. model.py +2 -1
model.py CHANGED
@@ -21,7 +21,8 @@ SAMPLES_PATH = "./samples/*.jpg"
21
 
22
  class ImageLoader():
23
  def __init__(self):
24
- self.transformer = torchvision.transforms.Compose([ImageTransformsFactory.create("global_resize::{'size':255}"),
 
25
  ImageTransformsFactory.create("center_crop::{'size':224}"),
26
  ImageTransformsFactory.create("normalize"),
27
  torchvision.transforms.ToTensor()])
 
21
 
22
  class ImageLoader():
23
  def __init__(self):
24
+ self.transformer = torchvision.transforms.Compose([torchvision.transforms.ToTensor(),
25
+ ImageTransformsFactory.create("global_resize::{'size':255}"),
26
  ImageTransformsFactory.create("center_crop::{'size':224}"),
27
  ImageTransformsFactory.create("normalize"),
28
  torchvision.transforms.ToTensor()])