Spaces:
Runtime error
Runtime error
syntax fix
Browse files
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([
|
|
|
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()])
|