google colaboratoryにおけるlightweight-ganの実装ができない

前提・実現したいこと

google colaboratory proにてlightweight-ganの実装を試みました。

lightweight-ganのインポートやgoogledriveのマウントまではスムーズに進んだのですが、GANを実行したところkorniaからfilter2dをインポートできないとのエラーメッセージが表示されました。

同様の環境にて、つい最近までは動いていたとの書き込みを多数発見することができましたが、当方はプログラミング初学者につき今回のエラーが何に起因するものなのかを判定することができませんでした。有識者の方がいらっしゃれば解決方法をご教授ください。

発生している問題・エラーメッセージ

Traceback (most recent call last):
  File "/usr/local/bin/lightweight_gan", line 5, in <module>
    from lightweight_gan.cli import main
  File "/usr/local/lib/python3.7/dist-packages/lightweight_gan/__init__.py", line 1, in <module>
    from lightweight_gan.lightweight_gan import LightweightGAN, Generator, Discriminator, Trainer, NanException
  File "/usr/local/lib/python3.7/dist-packages/lightweight_gan/lightweight_gan.py", line 25, in <module>
    from kornia import filter2d
ImportError: cannot import name 'filter2d' from 'kornia' (/usr/local/lib/python3.7/dist-packages/kornia/__init__.py)

該当のソースコード

python
1 !pip install lightweight-gan

python
1! lightweight_gan \
2    --data ./goh-jpeg \
3    --name "bi-face" \
4    --batch-size 16 \
5    --gradient-accumulate-every 4 \
6    --num-train-steps 200000

試したこと

インポート自体を別で実行しようと試みましたが、そのようなライブラリは存在しないと表示されてしまいました。

補足情報（FW/ツールのバージョンなど）

環境：google colaboratory　pro
言語：python３.７

jbpb0

2021/10/23 09:52 編集

Google Colabで、質問に掲載のエラーが再現しましたが、 !pip install kornia==0.5.11 を行なったら、エラー出なくなりましたエラーの原因は、kornia 0.6.0で仕様が変わって下記ができなくなったからです > from kornia import filter2d

kokinojima

2021/10/24 08:45

ご丁寧にありがとうございます！korniaのバージョンが上がってしまっていたのですね。ちなみになんですが、lightweight-ganをpipで入れた後にこちらのコマンドを実行するという認識であってますでしょうか？

jbpb0

2021/10/24 09:17

どちらが先でも、結果的には同じ状態になるので大丈夫ですが、 > lightweight-ganをpipで入れた後にこちらのコマンドを実行だと、「kornia」の最新版がまず入って、その後に !pip install kornia==0.5.11 で古いのに入れ替えることになります一方、先に !pip install kornia==0.5.11 を実行しておくと、 > lightweight-ganをpipで入れの時にはそれがそのまま使われるので、「kornia」のインストールは一回だけです後者の方が、インストールにかかるトータル時間が(わずかな差かもしれませんが)短くなると思います

kokinojima

2021/10/24 12:38

そのような違いがあるとは存じ上げませんでした。丁寧にありがとうございます！調べた限りではこのような情報はどこにも書いてなかったので大変助かりました。初めて質問したため、回答済みにすることや評価する方法がわからなくて恐縮です。