Google Colaboratory及び学習済みStyleGANによる人物画像生成

Pythonで画像処理してみたいです。

このような要望にお応えします。

今回は、StyleGANによる人物画像生成を行ってみます。

下記のサイトを参考にさせていただきました。

StyleGANについて

StyleGANのモデル、学習方法については、以下の記事がよいと思いました。

今回は、下記で公開されている学習済みstyleGANを用いて人物画像生成をします。

・Googleのアカウントを作成します。
・Googleドライブにアクセスし、「新規」→「その他」から「Google Colaboratory」の順でクリックします。そうすると、Colaboratoryが起動します。

・Colaboratoryが起動したら、以下のコマンドをCoalboratoryのセルに入力し実行します。
そうすることで、Googleドライブをマウントします。

from google.colab import drive
drive.mount('/content/drive')

1 2	from google.colab import drive drive.mount('/content/drive')

・実行後、認証コードの入力が促されます。このとき、「Go to this URL in a browser」が指しているURLにアクセスしgoogleアカウントを選択すると、認証コードが表示されますので、それをコピーしenterを押します。これでGoogleドライブのマウントが完了します。

ツールをダウンロードする場所に移動します。
本記事では、マイドライブにダウンロードします。

cd /content/drive/My Drive

1	cd /content/drive/My Drive

gitからツールをダウンロードします。

!git clone https://github.com/NVlabs/stylegan.git

1	!git clone https://github.com/NVlabs/stylegan.git

ダウンロードしたstyleganまで移動します。

cd stylegan

1	cd stylegan

styleganを使用するにあたり、tensorflowのバージョンを1.15.0にして実施しています。

styleganフォルダに含まれるpretrained_example.pyを用いて、学習済みStyleGANモデルによる人物画像を生成してみます。こちら(http://cedro3.com/ai/stylegan/)を参考に、出力結果をGIF画像として保存する処理を追加して出力しました。

!python pretrained_example.py

1	!python pretrained_example.py

どうでしょうか?今回は、StyleGANの学習済みモデルを使用して人物画像を生成してみました。StyleGANを用いることで、様々な画像編集ができそうですので試してみてはいかがでしょうか。