シンプルなGANの実装

GANの論文を読んだついでにMNIST用に実装しました.

ネットワークには,Convolutionを使わず,多層ニューラルネットです.層の数,活性化関数やバッチ正規化などの設定はてきとうです.なのでGANもどきです.

f:id:akmtn:20170514191216p:plainがんもどき

 

pytorchで実装したかったので,本家のDCGANのコードを参考にしながら書きました.

github.com

また,こちらの方の実装も参考にしました.(keras)

yusuke-ujitoko.hatenablog.com

 

結果として,以下のような出力を得ました.

f:id:akmtn:20170514190525p:plain

それなりに数字には見えますが,1が多い一方,2が少ないという偏りがありました.

 

一応,コードはここにあります.