pickerLab人生のオープンソース化。イラストpicker.あけましておめでとうございます!!今年もどうぞよろしくお願いします。皆さんとどういう形で関われるのかわかりませんが、来年もたまにこのブログを読んでいっていただけたら嬉しいです。牛のイラストを生成してみた画像を生成できるニューラルネットワークであるGNAなるものが世の中に存在するのは昔から知ってましたが、そういうものがあるって聞いていただけで、使ったこともないし、原理もよくわからないし・・・でもなんか派手だなぁーって思ってた感じです。GAN”ってモデルがすごいという話を多方面から聞くんですよね。コマンドだけで実行できる実装済みのモデルがgitで公開されているという。これは試してみるしかないと思ったわけです。思ったのが12月上旬でした。そこからなんの画像を学習させるか考えたのですが、まぁブログとか載せるとしたら今年の干支の牛のイラストでも生成すると思ったわけです。取り掛かったのが仕事納めをした後だったのであれやこれや工夫する時間はなくて、とりあえず大晦日に出力できた画像をphotoshopできれいにした画像がサムネになっている下の画像になります。最初に始めたのがデータ集めですが正直どんな画像が生成しやすいとかもわかってないです。匿名の開発者の論文にはポケモンやアニメキャラを生成した画像もサンプルとして掲載されていたので、牛のイラストも作れるよなぁとおもむろにデータを集めてgitからダウンロードしたモデルに突っ込みました。lucidrains/lightweight-gan原理をちゃんと理解していないモデルをお試しに使っているのでそんな複雑なことはしません。難易度を下げるために、画面上の牛のパーツの位置を固定しました。右側に顔、左側にお尻。シルエットかホルスタインのイラストのみを使いました。GAN”は100枚以上あれば動くそうなので牛イラストを130枚集めて、K80を積んだAzureのVMで夜通しモデルを学習させました。lightweight-gan`でインストールできます。その後はpythonでないかコードを書く必要はなく、lightweight-ganでモデルの学習から画像の生成までを行うことができます。100000`こんな具合です。すると、コンソールに進行状況が逐次出力されるようになり、1000イテレーションごとにチェックポイントとしてモデルと現在の出力画像がサンプルとして一枚の画面に64の画像が保存されます(下の画像は6000イテレーション時のサンプル画像)。結果あまり上手くいかなかった10000イテレーション学習させたあたりから、ある教師画像のイラストそっくりな画像が大量に生成されるようになった。一度学習を止めて、1000イテレーションづつのチェックポイントから再開したら、教師画像にチェックポイントごとに生成されるサンプル画像が教師画像に混ざってしまい8×8のサンプル画像が生成されるようになってしまった。6000イテレーション学習させたところでvmが止まってしまって、しょうがないのでチェックポイントから始めたのですが、それからいろいろおかしくなりました。同じ画像が大量に生成される。ここら辺のことを考察するにはまだ知識が足りてないのですが、もしかすると過学習のような現象が起きてきました。2,3枚の教師画像に酷似した画像がどんどん生成されるようになって、逆にそれ以外の画像は上手く生成できなくなりました。今思った原因は、教師画像が少なかったか教師画像に同じ画像が複数あったとかそんなところでしょうか?そもそも、学習の進め方やモデルの使い方に間違いがあったのかもしれないです。これに関してはgitのコマンドをコピペしてるので検証しようがないという感じです。上の画像のように教師データと同じ画像がいくつも生成されるんですよね。あともう一つ、サンプル画像が生成されちゃっています。サンプル画像が生成されるのを回避するには、学習を再開させる前に教師データのフォルダ中にあるdefaultというファイルの中にあるサンプル画像を削除することで回避できると思います(未検証)。モデルの実装をしてみたいと思ったもとから人のコードを利用させてもらうのが苦手というのもありますが、コマンドだけで実行しているとうまくいかないときにやれることが少ないので、勉強のためにも多少コードを書きたいと思いました。どのレベルで実装するかはよくわかりませんが、論文を片手にコードを書いてみたほうが当たり前ですけどその後に役立つ知見は得られますよね。初心者の僕が最初の一歩としてGANを触ってみるにはいいネタでしたね。来年の干支のトラで再チャレンジしてみようと思います。ことしは上手くいかなかったので人間が手書きにで書くことにしました(使いどころがあったらご自由にお使いください)。this...TwitterFacebookPinterestTumblrコメントを残すメールアドレスが公開されることはありません。ニューラルネットワークやってました。むむむと思った人はtwitterフォローお願いします。本サイトへのリンクは自由に貼ってください(批判もOKです)。こちらのサイトでコラムを書いてます。24年卒向け新卒採用実施しております。リンク先サイトに登録するとインターンなどの情報も流れてきます。