basicsrが新しめのtorchvisionでエラー

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

というエラーになるので調べてみるとだいぶ前からissueになってる様子

[BREAKING FIX] [torchvision 0.17] Change functional_tensor import by TornjV · Pull Request #650 · XPixelGroup/BasicSR · GitHub

issueはcloseとなっているが、PRを見るとopenedのままで、どうやらbackward compatibilityのためのちょっとした修正が反映されてなくてそのままっぽい

Fix import error for rgb_to_grayscale by sipsmehta · Pull Request #677 · XPixelGroup/BasicSR · GitHub

ただ、これだけなら別PRにして入れればよいだけなので何か別の問題があるのかもしれない

ひとまず個人的に使いたいのでforkしてこの修正だけ最新リリースの1.4.2に加えたブランチを作って使うことにした

https://github.com/y-kamiya/BasicSR/tree/fix/latest-torchvision

pip install git+https://github.com/y-kamiya/BasicSR.git@fix/latest-torchvision

でOK