🔰StyleGAN3を用いて画像生成を行いたい

実現したいこと

自身が持つオリジナルデータを学習させて、styleGAN3を通して新しい画像を生成させたい。
https://github.com/NVlabs/stylegan3
↑このサイトから読み解こうと頑張っています。

前提

環境
・wsl2
・ubuntu20
・cuda 11.6
・python3.9

実装方法がわからず苦戦しているため、認識のズレに対する訂正やアドバイスを頂きたいです。
現在は、dekktop上に以下のコードを実行して諸々をダウンロードしている状態です。

git clone https://github.com/NVlabs/stylegan3.git

その後、カレントディレクトリでこのstylegan3のフォルダに移動して実行しようと考えているところです。
以下フォルダの中身です。
イメージ説明

実行しようとしているコード

python dataset_tool.py --source=〇△× --dest=〇△× python train.py --outdir=〇△× --data=〇△× --gpus=1 python gen_images.py --outdir=〇△× --trunc=1.0 --seeds=0-99 --network=〇△×

ここで何点か質問があります。

①:--source部分は、自分が学習させたい画像(新しい画像を生成させるためのオリジナル画像)が保存されているディレクトリまでのパスを入力すればよいのでしょうか。
②:--dest部分は、dataset_tool.pyを使って前処理?などを行った画像を一時的に保存させるためのディレクトリまでのパスを指定すればよいのでしょうか。
③:gen_images.pyで記述のある--outdirには、生成した画像を保存するためのディレクトリのパスを指定すればよいのでしょうか。
④:--networkというのがあまりわからないのですが、既に学習して重みなどが保存されているjsonファイル(ネットに転がっている?)などへのパスを通せばよいのでしょうか。
⑤:train.pyで指定する、--dataは恐らく②部分で処理した画像がある--dest部分のパスと同じで良いと認識していますが、--outdirはどのようなものを指定すればよいのか検討つきません。
⑥:また任意の枚数分生成させたい時は、--seedsの部分の数字を0-999(1000枚分)とかにすればよいのでしょうか。

まだ実行できておらず認識に齟齬がないか有識者の方にお聞きしたく投稿しています。
アドバイスいただければ幸いです。

補足情報

#生成AI #画像処理 #GAN

コメントを投稿

0 コメント