From 29cbe433531da8c75ed1a1cf2137e580cce03acd Mon Sep 17 00:00:00 2001 From: Gordon Hui <125633533+gordon0907@users.noreply.github.com> Date: Sat, 3 Jun 2023 00:37:43 +0900 Subject: [PATCH] Update degradations.py Change import source of `rgb_to_grayscale` from `torchvision.transforms.functional_tensor` to `torchvision.transforms.functional` --- basicsr/data/degradations.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/basicsr/data/degradations.py b/basicsr/data/degradations.py index 4fa20d4a0..14319605d 100644 --- a/basicsr/data/degradations.py +++ b/basicsr/data/degradations.py @@ -5,7 +5,7 @@ import torch from scipy import special from scipy.stats import multivariate_normal -from torchvision.transforms.functional_tensor import rgb_to_grayscale +from torchvision.transforms.functional import rgb_to_grayscale # -------------------------------------------------------------------- # # --------------------------- blur kernels --------------------------- #