Spaces:
Runtime error
Runtime error
Davidzhangyuanhan
commited on
Commit
·
43382ea
1
Parent(s):
811c881
Add application file
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def pil_loader(filepath):
|
|
22 |
|
23 |
def build_transforms(input_size, center_crop=True):
|
24 |
transform = torchvision.transforms.Compose([
|
25 |
-
torchvision.transforms.ToPILImage()
|
26 |
torchvision.transforms.Resize(input_size * 8 // 7),
|
27 |
torchvision.transforms.CenterCrop(input_size),
|
28 |
torchvision.transforms.ToTensor(),
|
|
|
22 |
|
23 |
def build_transforms(input_size, center_crop=True):
|
24 |
transform = torchvision.transforms.Compose([
|
25 |
+
torchvision.transforms.ToPILImage(),
|
26 |
torchvision.transforms.Resize(input_size * 8 // 7),
|
27 |
torchvision.transforms.CenterCrop(input_size),
|
28 |
torchvision.transforms.ToTensor(),
|