diff --git a/timm/data/transforms_factory.py b/timm/data/transforms_factory.py index a5facbf5..3fb73272 100644 --- a/timm/data/transforms_factory.py +++ b/timm/data/transforms_factory.py @@ -145,6 +145,7 @@ def transforms_imagenet_eval( scale_size = tuple([int(x / crop_pct) for x in img_size]) else: scale_size = int(math.floor(img_size / crop_pct)) + scale_size = (scale_size, scale_size) tfl = [ transforms.Resize(scale_size, interpolation=str_to_interp_mode(interpolation)),