Não consigo fazer oque o professor fez na aula

[Sandro Eduardo Prado Volski]

Segui passo a passo do que o professor fez em aula, porém quando chegou na célula do código onde escrevemos:
!python inference_gfpgan.py -i inputs/testes -o results -v 1.3 -s 2 –bg_upsampler realesrgan

O Google Colab, retorna o seguinte:
Traceback (most recent call last): File “/content/GFPGAN/GFPGAN/inference_gfpgan.py”, line 7, in <module> from basicsr.utils import imwrite File “/usr/local/lib/python3.10/dist-packages/basicsr/__init__.py”, line 4, in <module> from .data import * File “/usr/local/lib/python3.10/dist-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 “/usr/local/lib/python3.10/dist-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 “/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 <module> 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 <module> from torchvision.transforms.functional_tensor import rgb_to_grayscale ModuleNotFoundError: No module named ‘torchvision.transforms.functional_tensor’

Pensei que fosse alguma incompatibilidade de versões, porém já tentei atualizar e mesmo assim continuo com este erro, o que devo fazer ? Pois nem no material que é disponibilizado pelo curso este trecho do código não está funcionando corretamente

[Gabriel Alves]

Olá Sandro!

Esse erro é devido a uma incompatibilidade recente que ocorreu com a versão automaticamente atribuída para o pytorch na sessão.

Para evitar essa mensagem de erro, basta executar o comando abaixo logo no início do Colab, executando toda vez que iniciar a sessão

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

Em seguida, pode continuar a rodar o restante do código.

Basta colar no início do Colab e executar, mas se tiver dúvidas confira aqui o Colab oficial da aula pois lá já se encontra atualizado.

[Autor]

Posts