We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 1468e91 + 1ccf536 commit 3363156Copy full SHA for 3363156
utils.py
@@ -128,7 +128,7 @@ def update(self, value):
128
129
def get_transform(target_size, central_fraction=1.0):
130
return transforms.Compose([
131
- transforms.Scale(int(target_size / central_fraction)),
+ transforms.Resize(int(target_size / central_fraction)),
132
transforms.CenterCrop(target_size),
133
transforms.ToTensor(),
134
transforms.Normalize(mean=[0.485, 0.456, 0.406],
0 commit comments