Skip to content

Commit 1ccf536

Browse files
committed
Scale -> Resize
1 parent 1468e91 commit 1ccf536

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)