【Colab】Module Not Found 'torchvision.transforms.functional_tensor'

Google Colabを使用してReal ESRGANを用いた動画高画質化を試しているとエラーが発生。

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

もしcolabを使っていないのであればlib/python3.10/site-packages/basicsr/data/degradations.pyの

from torchvision.transforms.functional_tensor import rgb_to_grayscale

↓

from torchvision.transforms.functional import rgb_to_grayscale

にするだけで改善されるらしいのですが、私はcolabを使用しているので別の解決方法を探しました。

そして、私の場合だとこのコマンドで改善できました：

!pip uninstall torch torchvision -y
!pip install torch==1.13.0 torchvision==0.14.0 -f https://download.pytorch.org/whl/cu121/torch_stable.html

これをcolabのコードブロックで実行すると：

/content/Real-ESRGAN
Enhancing 2575 Frames with ESRGAN...
Downloading: "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth" to /content/Real-ESRGAN/weights/RealESRGAN_x4plus.pth

100% 63.9M/63.9M [00:00<00:00, 338MB/s]

Testing 0 frame_00000001
Testing 1 frame_00000002
.
.
.

無事動きました！

もし躓いている方がいたら参考にしてください！