Pythonで画像処理してみたいです。
このような要望にお応えします。
今回は、StyleGANによる人物画像生成を行ってみます。
下記のサイトを参考にさせていただきました。
- https://github.com/NVlabs/stylegan
- https://qiita.com/pacifinapacific/items/1d6cca0ff4060e12d336
- https://www.slideshare.net/KentoDoi/stylegan-cvpr2019dena
- http://cedro3.com/ai/stylegan/
StyleGANについて
StyleGANのモデル、学習方法については、以下の記事がよいと思いました。
- https://qiita.com/pacifinapacific/items/1d6cca0ff4060e12d336
- http://cedro3.com/ai/stylegan/
- https://medium.com/@akichan_f/ganの基礎からstylegan2まで-dfd2608410b3
今回は、下記で公開されている学習済みstyleGANを用いて人物画像生成をします。
- https://github.com/NVlabs/stylegan
Google Colaboratoryの準備
・Googleのアカウントを作成します。
・Googleドライブにアクセスし、「新規」→「その他」から「Google Colaboratory」の順でクリックします。そうすると、Colaboratoryが起動します。
・Colaboratoryが起動したら、以下のコマンドをCoalboratoryのセルに入力し実行します。
そうすることで、Googleドライブをマウントします。
|
1 2 |
from google.colab import drive drive.mount('/content/drive') |
・実行後、認証コードの入力が促されます。このとき、「Go to this URL in a browser」が指しているURLにアクセスしgoogleアカウントを選択すると、認証コードが表示されますので、それをコピーしenterを押します。これでGoogleドライブのマウントが完了します。
StyleGAN学習済みモデルの準備
ツールをダウンロードする場所に移動します。
本記事では、マイドライブにダウンロードします。
|
1 |
cd /content/drive/My Drive |
gitからツールをダウンロードします。
|
1 |
!git clone https://github.com/NVlabs/stylegan.git |
ダウンロードしたstyleganまで移動します。
|
1 |
cd stylegan |
styleganを使用するにあたり、tensorflowのバージョンを1.15.0にして実施しています。
styleganフォルダに含まれるpretrained_example.pyを用いて、学習済みStyleGANモデルによる人物画像を生成してみます。こちら(http://cedro3.com/ai/stylegan/)を参考に、出力結果をGIF画像として保存する処理を追加して出力しました。
|
1 |
!python pretrained_example.py |
出力結果
どうでしょうか?今回は、StyleGANの学習済みモデルを使用して人物画像を生成してみました。StyleGANを用いることで、様々な画像編集ができそうですので試してみてはいかがでしょうか。
