Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
Matthew-3 opened this issue Apr 13, 2024 · 2 comments

Comments

@Matthew-3
Copy link

Anyone knows what's happened?
C:\Users\......\Desktop\tmp\tmp>python inference_realesrgan.py -n RealESRGAN_x4plus -i inputs --face_enhance Traceback (most recent call last): File "C:\Users\......\Desktop\tmp\tmp\inference_realesrgan.py", line 5, in <module> from basicsr.archs.rrdbnet_arch import RRDBNet File "D:\Python\Python311\Lib\site-packages\basicsr\__init__.py", line 4, in <module> from .data import * File "D:\Python\Python311\Lib\site-packages\basicsr\data\__init__.py", line 22, in <module> _dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "D:\Python\Python311\Lib\site-packages\basicsr\data\__init__.py", line 22, in <listcomp> _dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "D:\Python\Python311\Lib\importlib\__init__.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "D:\Python\Python311\Lib\site-packages\basicsr\data\realesrgan_dataset.py", line 11, in <module> from basicsr.data.degradations import circular_lowpass_kernel, random_mixed_kernels File "D:\Python\Python311\Lib\site-packages\basicsr\data\degradations.py", line 8, in <module> from torchvision.transforms.functional_tensor import rgb_to_grayscale ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor'

@o7451
Copy link

o7451 commented Apr 15, 2024

alter to torchvision.transforms. “_” functional_tensor
torchvision.transforms._functional_tensor

@VIPJoey
Copy link

VIPJoey commented May 1, 2024

alter to torchvision.transforms. “_” functional_tensor torchvision.transforms._functional_tensor

That's help for me.

vim /root/miniconda3/envs/ESRGAN/lib/python3.8/site-packages/basicsr/data/degradations.py

and then replace these code
from torchvision.transforms.functional_tensor import rgb_to_grayscale
to
from torchvision.transforms._functional_tensor import rgb_to_grayscale

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants