やりたいこと

今、はやりのDCGANの中身については、以前個人的に勉強してなんとなく中身は理解できているので、
ちょっと使ってみたいけど、実装を自前でするのはつらい...
とか思ってたら、chainerさんのgithubのexampleの中にあるではないか！？

これは使ってみるしかない！と思い使ってみたのと、少し自分なりに使いやすいように
改良(いや、改悪？）してみました。

DCGANについては、いろいろと解説しているページがあるのでそちらを見てください

DCGANの学習

chainerさんのインストールをして、学習
自前のデータセットはchainerのバージョンが1.2とかそんくらいに
「あ」「い」「う」「え」「お」の認識をしようと思って作ったものを利用
python -h train_dcgan.pyとすると、-iオプションでデータセットを指定できる！すばらしい

ということで、早速開始
python train_dcgan.py -g 0 -i ../mydataset/ --snapshot_interval 500 --display_interval 100

実行すると、なんと今、全体のうちどれくらい学習が終わっているかとか、
あとどのくらいで学習が終わりそうかなどが出力される
これまた感動...!

そして、学習が終わるとresult/previewフォルダに画像が

おおお！
上記、ちょっとソースをのぞいてみると、どうやら下記の部分で10×10のイメージ画像を、
一定のインターバルで出力している模様

101     trainer.extend(
102         out_generated_image(
103             gen, dis,
104             10, 10, args.seed, args.out),
105         trigger=snapshot_interval)

ということは、out_generated_imageを参考にすればできそう！
ということで、Let's try!

学習したGeneratorから任意の枚数の画像を個別に出力する

コードはgithubにコードおいてます。

https://github.com/komorin0521/chainer_myexample/tree/master/dcgan_generator

まず、generatorは、train_dcgan.pyの-rのコードを参考に、Generatorをロードするようにしてます。
出力はサンプルのvisualize.pyを参考にさせてもらいました。
そうすると、任意の枚数の画像が生成できた！
下記、サンプルです。

うおおお！すごい。と感動しっぱなしでした。笑

参考

chainer
https://github.com/chainer/chainer