TencentARC/GFPGAN

PLEASE HELP ANYONE - THIS ERROR (ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor')

Opened this issue · 5 comments

Traceback (most recent call last):
File "E:\SOFT\GFPGAN\GFPGAN\inference_gfpgan.py", line 7, in
from basicsr.utils import imwrite
File "C:\Users\MKN\anaconda3\Lib\site-packages\basicsr_init_.py", line 4, in
from .data import *
File "C:\Users\MKN\anaconda3\Lib\site-packages\basicsr\data_init_.py", line 22, in
dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames]
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MKN\anaconda3\Lib\site-packages\basicsr\data_init
.py", line 22, in
dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames]
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MKN\anaconda3\Lib\importlib_init
.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MKN\anaconda3\Lib\site-packages\basicsr\data\realesrgan_dataset.py", line 11, in
from basicsr.data.degradations import circular_lowpass_kernel, random_mixed_kernels
File "C:\Users\MKN\anaconda3\Lib\site-packages\basicsr\data\degradations.py", line 8, in
from torchvision.transforms.functional_tensor import rgb_to_grayscale


Uploading bdbe9c6d3f2e42f9f79e56e414fe27daed6c1ed0..jpg…

Have you managed to fix it? I'm getting the same when I run the script

@misssunitaparker @J-McFc

Go to the file degradations.py. It's located in the error message. For example, for @misssunitaparker, it's:

File "C:\Users\MKN\anaconda3\Lib\site-packages\basicsr\data\degradations.py"

Open the file degradations.py and edit line 8 from:

from torchvision.transforms.functional_tensor import rgb_to_grayscale

to:

from torchvision.transforms.functional import rgb_to_grayscale

Hopefully this works.

@DNucX It works, you're a legend, thanks mate.


ModuleNotFoundError Traceback (most recent call last)
Cell In[18], line 3
1 from pytorchvideo.data import LabeledVideoDataset, Kinetics, make_clip_sampler
----> 3 from pytorchvideo.transforms import (ApplyTransformToKey, Normalize, RandomShortSideScale, UniformTemporalSubsample,
4 )
6 from torchvision.transforms import (Compose, Lambda, RandomCrop, RandomHorizontalFlip, Resize)
8 from torchvision.transforms._transforms_video import (CenterCropVideo, NormalizeVideo)

File c:\Users\pnanurdb\AppData\Local\Programs\Python\Python312\Lib\site-packages\pytorchvideo\transforms_init_.py:3
1 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
----> 3 from .augmix import AugMix # noqa
4 from .mix import CutMix, MixUp, MixVideo # noqa
5 from .rand_augment import RandAugment # noqa

File c:\Users\pnanurdb\AppData\Local\Programs\Python\Python312\Lib\site-packages\pytorchvideo\transforms\augmix.py:6
3 from typing import Any, Dict, Optional
5 import torch
----> 6 from pytorchvideo.transforms.augmentations import (
7 _AUGMENTATION_MAX_LEVEL,
8 AugmentTransform,
9 _decreasing_int_to_arg,
10 _decreasing_to_arg,
11 _increasing_magnitude_to_arg,
12 _increasing_randomly_negate_to_arg,
...
----> 9 import torchvision.transforms.functional_tensor as F_t
10 from torchvision.transforms.functional import InterpolationMode
13 # Maximum global magnitude used for video augmentation.

ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor'

Can anyone help me with this error also? Thanks in advanced

@dinihanafi change torchvision.transforms.functional_tensor to torchvision.transforms.functional