From c5f7c3195d121c2b22bca2da8741a633fddb02a1 Mon Sep 17 00:00:00 2001 From: chuanli11 Date: Wed, 11 May 2022 08:12:32 -0700 Subject: [PATCH] fix get_image_size --- taming/data/image_transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/taming/data/image_transforms.py b/taming/data/image_transforms.py index 657ac332..661a868c 100644 --- a/taming/data/image_transforms.py +++ b/taming/data/image_transforms.py @@ -5,7 +5,7 @@ import torch from torch import Tensor from torchvision.transforms import RandomCrop, functional as F, CenterCrop, RandomHorizontalFlip, PILToTensor -from torchvision.transforms.functional import _get_image_size as get_image_size +from torchvision.transforms.functional import get_image_size as get_image_size from taming.data.helper_types import BoundingBox, Image