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.
1 parent 1468e91 commit 1ccf536Copy full SHA for 1ccf536
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