はじめに
ネット用に顔写真を載せなきゃいけなくなったけど、ネットに個人情報なんて載せたくないということがありますよね。
そこで学習済みのstylegan2を使って自分の画像に近い存在しない画像を簡単に生成できるようにしたものをgoogle colablatoryに公開しました。
google colablatoryとはgoogleが提供するもので、googleアカウントさえあれば無料でブラウザ上でGPUを使える神サービスです。
作ったものがこちらになります。
どんな画像が作れるの？
上から
- 元画像
- 元画像を切り取ったもの
- 生成された存在しない顔
となっています。
stylegan2のすごいところは1024x1024の高画質画像を生成できるところです。
また、おまけでgif動画も作れます。
元となったコード
このコードはstylegan2を試すものであり、
cedro様（@jun40vn）の「GANの潜在空間に新垣結衣は住んでいるのか？」
の記事内の「Search_for_Yui.ipynb」
をベースとして、自分の写真で実行しやすいように改良したものです。
@jun40vn 様の公開されたColab Notebooksがなければ作れませんした。心より感謝申し上げます。
改良点ついては適宜Colab Notebooks内で説明していますが、重要な部分は、
- 画像をアップロードできるようにした
- 探索回数を指定できるようにした
- 探索中の画像の保存回数を指定できるようにした。
です。
また写真はすべて
フリー素材ぱくたそ（ www.pakutaso.com ）
から使用しています。
使い方
1.まず公開したColab Notebooksにアクセスします。（googleアカウントにログインしてください。）
2.実行する
上部の「ランタイム」→「すべてのセルを実行」をクリック
3.画像をアップロードする
画像のアップロードという項に、以下のような「ファイルの選択」が現れます。
なので生成したい画像の素となる画像をアップロードしてください。
このようにドラックアンドドロップすることもできます。
4.探索を待つ
探索に一番時間がかかります。
気長に待ってください。
5.画像の表示
学習が終わったら順次画像が表示されていきます。
元画像と最終的な結果の表示
上から
- 元画像
- 元画像を切り取ったもの
- 生成された存在しない顔
となっています。
探索中の生成画像の変化
ここで、左に行くほど初期の生成画像となっています。
一番右は元となった画像なので注意してください。
高画質画像
1024x1024の高画質画像が表示されます。
右クリックで保存できます。
6.おまけ
パラメータの変化
探索回数をnum_stepsで変更できます。（デフォルトで300）
大きくするとよりアップロードした画像に近づきますが、実行時間が長くなります。
探索中の画像の表示回数はnumber_snapshotsで変更できます。（デフォルトで5）
大きくしてもあまり実行時間が変わったりしませんが、大きすぎると割り当てられたRAMやディスクを使い果たしてしまうかもしれません。
GPU
使用するGPUについては、Colab Notebooksの上のほうに表示されています。
詳しくは「google colab GPU ガチャ」で検索
Tesla T4がおそらく当たりです。
gif
複数枚アップロードした場合は、順番に変化していくgif動画を作れます。
注意
表示範囲
今回のプログラムでは、目の位置が中心になるように画像を切り取っています。（うろ覚え、詳しくは元となった記事を読んでください。）
そのため、場合によっては頭の上が切れてしまうことがあります
複数人
複数人写っていると人数分切り取られます。
しかし私の改良したプログラムでは、アップロードした枚数分しか学習やら表示やらをしません。
また顔でない部分が顔と認識されてしまうこともあります。
そこらへんはうまくやってください。
顔が認識されない
顔が認識されないこともありますがどうしようもありません。
他の画像を使用してください。
まとめ
写真に近い存在しない画像を生成する方法を説明しました。
ぜひ試してみてください。
そして改めて、@jun40vn 様の公開されたColab Notebooksがなければ作れませんした。心より感謝申し上げます。