diff --git a/data/base_dataset.py b/data/base_dataset.py index 2d93c9b..ea2fad9 100755 --- a/data/base_dataset.py +++ b/data/base_dataset.py @@ -37,7 +37,7 @@ def get_transform(opt, params, method=Image.BICUBIC, normalize=True, grayscale=F transform_list = [] if 'resize' in opt.resize_or_crop: osize = [opt.loadSize, opt.loadSize] - transform_list.append(transforms.Scale(osize, method)) + transform_list.append(transforms.Resize(osize, method)) elif 'scale_width' in opt.resize_or_crop: transform_list.append(transforms.Lambda(lambda img: __scale_width(img, opt.loadSize, method)))