StyleGAN2をGoogle Colaboratoryで実装したい

Question

前提・実現したいこと

Google ColaboratoryでStyleGAN2を実装しようとしています。(初学者です)
imagenet を用いて実装し、画像Aから画像Bへのトランジションを行いたいです。

主に以下のサイトを参考にして進めています。
・https://techpr.info/ml/stylegan2-colab/
・https://github.com/justinpinkney/awesome-pretrained-stylegan2

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

Local submit - run_dir: results/00000-generate-images
dnnlib: Running run_generator.generate_images() on localhost...
Loading networks from "gdrive:sdc/stylegan2-imagenet-512/model.ckpt-533504.pkl"...
Traceback (most recent call last):
  File "/content/stylegan2/run_generator.py", line 168, in <module>
    main()
  File "/content/stylegan2/run_generator.py", line 163, in main
    dnnlib.submit_run(sc, func_name_map[subcmd], **kwargs)
  File "/content/stylegan2/dnnlib/submission/submit.py", line 343, in submit_run
    return farm.submit(submit_config, host_run_dir)
  File "/content/stylegan2/dnnlib/submission/internal/local.py", line 22, in submit
    return run_wrapper(submit_config)
  File "/content/stylegan2/dnnlib/submission/submit.py", line 280, in run_wrapper
    run_func_obj(**submit_config.run_func_kwargs)
  File "/content/stylegan2/run_generator.py", line 21, in generate_images
    _G, _D, Gs = pretrained_networks.load_networks(network_pkl)
  File "/content/stylegan2/pretrained_networks.py", line 72, in load_networks
    stream = open(path_or_url, 'rb')
FileNotFoundError: [Errno 2] No such file or directory: 'gdrive:sdc/stylegan2-imagenet-512/model.ckpt-533504.pkl'

該当のソースコード

# tensorflow
!pip uninstall tensorflow -y
!pip install tensorflow-gpu==1.15.0
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

# カレントディレクトリからでもStyleGAN２のモジュールを使えるように設定
import sys
sys.path.insert(0, "/content/stylegan2")
!git clone https://github.com/NVlabs/stylegan2.git

# 画像生成の実行
!python /content/stylegan2/run_generator.py generate-images \
    --network=gdrive:sdc/stylegan2-imagenet-512/model.ckpt-533504.pkl \
  --seeds=6600-6625 --truncation-psi=0.5

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

stylegan2-car-config-f.pklと同じようにダウンロードリンクをコード内に記述したのですがどこをどう変えるのかが分からず行き詰まっています。
下記のがダウンロードリンクです。

car-config-f
https://nvlabs-fi-cdn.nvidia.com/stylegan2/networks/stylegan2-car-config-f.pkl

ImageNet
https://battle.shawwn.com/sdc/stylegan2-imagenet-512/model.ckpt-533504.pkl

Comments

[meg_]

!python /content/stylegan2/run_generator.py generate-images \
--network=gdrive:sdc/stylegan2-imagenet-512/model.ckpt-533504.pkl \
--seeds=6600-6625 --truncation-psi=0.5

上記コードはどこからの引用ですか？

[jbpb0]

下記は普通は、tensorflow 2.*を1.*のように動かしたい場合に使うもので、tensorflow 1.*を使う場合は不要だと思います

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

[jbpb0]

「!python...」と別プロセスでpythonを実行する場合は、たぶん下記は効果無いと思います

import sys
sys.path.insert(0, "/content/stylegan2")

Answer 1

画像生成の実行
!python /content/stylegan2/run_generator.py generate-images
--network=gdrive:sdc/stylegan2-imagenet-512/model.ckpt-533504.pkl
--seeds=6600-6625 --truncation-psi=0.5

model.ckpt-533504.pklを使用したいのであれば下記コードで実行できるかと思います。

!wget https://battle.shawwn.com/sdc/stylegan2-imagenet-512/model.ckpt-533504.pkl
# 画像生成の実行
!python /content/stylegan2/run_generator.py generate-images \
    --network=model.ckpt-533504.pkl \
  --seeds=6600-6625 --truncation-psi=0.5

Answer 2

質問のコードを下記のように変更してgoogle colabで実行したら、

FileNotFoundError: [Errno 2] No such file or directory: 'gdrive:sdc/stylegan2-imagenet-512/model.ckpt-533504.pkl'

は出なくなりました

# tensorflow
!pip uninstall tensorflow -y
!pip install tensorflow-gpu==1.15.0
#import tensorflow.compat.v1 as tf
#tf.disable_v2_behavior()

# カレントディレクトリからでもStyleGAN２のモジュールを使えるように設定
#import sys
#sys.path.insert(0, "/content/stylegan2")
!git clone https://github.com/NVlabs/stylegan2.git

# 画像生成の実行
#!python /content/stylegan2/run_generator.py generate-images \
#    --network=gdrive:sdc/stylegan2-imagenet-512/model.ckpt-533504.pkl \
#  --seeds=6600-6625 --truncation-psi=0.5
!wget https://battle.shawwn.com/sdc/stylegan2-imagenet-512/model.ckpt-533504.pkl
!python /content/stylegan2/run_generator.py generate-images \
    --network=model.ckpt-533504.pkl \
    --seeds=6600-6625 --truncation-psi=0.5

Comments

[jbpb0]

「質問への追記・修正の依頼」に書いた、質問のエラーとは直接関係無いところも直してます

[riku_university]

上記に記載してくださったコードで実行することができました！
もう一人の方のほうが早かったためそちらの方をBAにさせていただきました。
解答していただきありがとうございました。