• Stars
    star
    933
  • Rank 48,644 (Top 1.0 %)
  • Language
    Python
  • License
    MIT License
  • Created almost 9 years ago
  • Updated about 4 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Chainer implementation of Deep Convolutional Generative Adversarial Network

You can try web demo here !!

chainer-DCGAN

Chainer implementation of Deep Convolutional Generative Adversarial Network (http://arxiv.org/abs/1511.06434)

説明

画像を生成するニューラルネットです。
12/24のchainer advent calendarに解説を書きました。 http://qiita.com/mattya/items/e5bfe5e04b9d2f0bbd47
このコードは現在試行錯誤の途中であり、突然の変更などの可能性が十分あります。ご了承ください。

使い方(暫定)

  • chainer 1.5が必要
  • 学習済みモデルから生成のみを行うには、visualizer.pyを使用する。GPU無くてもOK。 python visualizer.py
  • 学習を行うにはDCGAN.pyを実行する。image_dir変数で指定されたディレクトリに、学習元となる画像ファイルを置く。GPUが必要で、何時間かかかる。

サンプル

20万枚の顔イラスト画像で約3時間学習を行った結果(GTX 970使用)。

特定の画像の生成元となったベクトルzにノイズを加えると、髪型や服装などが少しずつ異なる画像を生成できる。 このことから、本モデルが過学習しているわけではない(特定の画像を暗記しているわけではない)ことが示唆される。

画像間の連続的変換。

参考文献

本家の実装です。モデルの相違点はleaky_reluの代わりにeluを使っているくらいです。 https://github.com/soumith/dcgan.torch