Skip to content

Commit 3363156

Browse files
authored
Merge pull request #25 from gipster/scale-to-resize
Changes transform.Scale to transforms.Resize
2 parents 1468e91 + 1ccf536 commit 3363156

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def update(self, value):
128128

129129
def get_transform(target_size, central_fraction=1.0):
130130
return transforms.Compose([
131-
transforms.Scale(int(target_size / central_fraction)),
131+
transforms.Resize(int(target_size / central_fraction)),
132132
transforms.CenterCrop(target_size),
133133
transforms.ToTensor(),
134134
transforms.Normalize(mean=[0.485, 0.456, 0.406],

0 commit comments

Comments
 (0)