xinntao / Real-ESRGAN

Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.
BSD 3-Clause "New" or "Revised" License
27.88k stars 3.51k forks source link

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

Open Haremakhet opened 6 months ago

Haremakhet commented 6 months ago

Traceback (most recent call last): File "/content/Real-ESRGAN/inference_realesrgan.py", line 5, in from basicsr.archs.rrdbnet_arch import RRDBNet File "/usr/local/lib/python3.10/dist-packages/basicsr/init.py", line 4, in from .data import * File "/usr/local/lib/python3.10/dist-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 "/usr/local/lib/python3.10/dist-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 "/usr/lib/python3.10/importlib/init.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "/usr/local/lib/python3.10/dist-packages/basicsr/data/realesrgan_dataset.py", line 11, in from basicsr.data.degradations import circular_lowpass_kernel, random_mixed_kernels File "/usr/local/lib/python3.10/dist-packages/basicsr/data/degradations.py", line 8, in from torchvision.transforms.functional_tensor import rgb_to_grayscale ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor'

RohitPoul commented 6 months ago

go to the file show here "/usr/local/lib/python3.10/dist-packages/basicsr/data/degradations.py" do this change in the code - torchvision.transforms.functional import rgb_to_grayscale you have to replace it every single use

tushar-31093 commented 6 months ago

go to the file show here "/usr/local/lib/python3.10/dist-packages/basicsr/data/degradations.py" do this change in the code - torchvision.transforms.functional import rgb_to_grayscale you have to replace it every single use

So does this imply that this repo wont function anymore without explicity changing this? At the same time colab wont work either?

viniciusmods commented 6 months ago

`!nvidia-smi

from PIL import Image import cv2, os, subprocess from google.colab import drive from tqdm import tqdm

!git clone https://github.com/xinntao/Real-ESRGAN.git %cd Real-ESRGAN

with open('requirements.txt', 'w') as arquivo: arquivo.write("basicsr==1.3.5\ngfpgan>=1.3.5\nnumpy\nopencv-python\nPillow\ntorch>=1.7\ntorchvision\ntqdm\n")

!pip install facexlib ffmpeg ffmpeg-python !pip install -r requirements.txt !python setup.py develop

with open('/usr/local/lib/python3.10/dist-packages/basicsr-1.3.5-py3.10.egg/basicsr/data/degradations.py', 'r') as file: file_data = file.read() file_data = file_data.replace('from torchvision.transforms.functional_tensor import rgb_to_grayscale', 'from torchvision.transforms.functional import rgb_to_grayscale') with open('/usr/local/lib/python3.10/dist-packages/basicsr-1.3.5-py3.10.egg/basicsr/data/degradations.py', 'w') as file: file.write(file_data)

mount_drive=False #@param{type:"boolean"}

if mount_drive: drive.mount('/content/gdrive/')`

JossCamp commented 6 months ago

I don't know why in Colab, whenever something gets updated, it always has to mess up everything else. To solve this problem, you need to execute this command.

!python -m pip install torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118

After this, everything should work fine.

rahulmahmud commented 6 months ago

go to the file show here "/usr/local/lib/python3.10/dist-packages/basicsr/data/degradations.py" do this change in the code - torchvision.transforms.functional import rgb_to_grayscale you have to replace it every single use

please give a proper video

eharsh2711 commented 6 months ago

I don't know why in Colab, whenever something gets updated, it always has to mess up everything else. To solve this problem, you need to execute this command.

!python -m pip install torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118

After this, everything should work fine.

Can confirm, this worked for me in colab.

m4ra7h0n commented 5 months ago

I don't know why in Colab, whenever something gets updated, it always has to mess up everything else. To solve this problem, you need to execute this command.

!python -m pip install torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118

After this, everything should work fine.

Can confirm, this worked for me in colab.

ERROR: Could not find a version that satisfies the requirement torch==2.0.1 (from versions: 2.2.0, 2.2.1, 2.2.2) ERROR: No matching distribution found for torch==2.0.1