1時間で画像生成の環境構築(Keras-GAN)

Notionでメモを取っていたら、意外とまとまったので投稿してみる。

Intentions

  1. 環境構築
    1. GPUライブラリ(CUDA, cudnn)
    2. Pythonライブラリ(keras, tensorflow)
  2. Keras-GANの実行

Happenings

バージョンを合わせないと駄目っぽい。

最終的にPython3.7、TF1.14.0、CUDA10.0、cudnn7.6.0で落ち着いた。これはKeras-GANの環境に限った環境なので、コードごとに環境を整える必要がありそう。

ACGANの学習を実行した。10分もかからずに終了。MNISTサンプルではこんなものか。

次はとりあえず、カントクデータをface detectionして学習する。

結果のログ

epoch 200

f:id:wakadori_Mk2:20200401194801p:plain
epoch 200

epoch 2000

f:id:wakadori_Mk2:20200401194833p:plain
epoch 2000

epoch 14000

f:id:wakadori_Mk2:20200401194836p:plain
epoch 14000

Grateful for

  1. 久しぶりにGANをぶん回したので精神が安定した。
  2. 環境構築を1時間で済ませられた、えらい。

Action items

  • [ ] face detectionの実装。
  • [ ] 入力データ形式に変換する。