diff --git a/basicsr/data/degradations.py b/basicsr/data/degradations.py index 14319605d..892826b26 100644 --- a/basicsr/data/degradations.py +++ b/basicsr/data/degradations.py @@ -5,7 +5,10 @@ import torch from scipy import special from scipy.stats import multivariate_normal -from torchvision.transforms.functional import rgb_to_grayscale +try: + from torchvision.transforms.functional import rgb_to_grayscale +except ImportError: + from torchvision.transforms.functional_tensor import rgb_to_grayscale # -------------------------------------------------------------------- # # --------------------------- blur kernels --------------------------- #