StyleGAN2をGoogle Colaboratoryで実装したい

前提・実現したいこと

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

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

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

Python

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'

該当のソースコード

Python

# tensorflow!pip uninstall tensorflow -y !pip install tensorflow-gpu==1.15.0import tensorflow.compat.v1 as tf tf.disable_v2_behavior() # カレントディレクトリからでもStyleGAN2のモジュールを使えるように設定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

コメントを投稿

0 コメント