diff --git a/transforms.py b/transforms.py index 7ccbab8..dbab259 100644 --- a/transforms.py +++ b/transforms.py @@ -120,7 +120,7 @@ def get_params(img, scale, ratio): tuple: params (i, j, h, w) to be passed to ``crop`` for a random sized crop. """ - width, height = FT._get_image_size(img) + width, height = FT.get_image_size(img) area = height * width log_ratio = torch.log(torch.tensor(ratio))