snjstudent / CycleGAN.tensorflow.keras

動かしてみたくなったので、CycleGANを実装

Abstract

CycleGANをtensorflow.keras独特のModelとLayerを使い実装。
CycleGAN:https://arxiv.org/abs/1703.10593

CycleGAN.py

ローカル上で動かす場合のファイル。画像を読み込むディレクトリはimage_domainA内と、image_domainB内のファイル

CycleGAN_colab.py

Google colabolatoryで動かす時用のファイル。
同様のディレクトリをGoogle Drive内に入れることで実行可能。
ミニバッチ学習を行っており、全体において５stepごとに重み保存。
tensorflow_addonがインストールできなかったため、そこだけ直接コピペしています。
Discriminatorの学習速度がGeneratorに比べ早いので、Discriminatorは３回に1回しか学習しないようにしています。

data_agumentation.py

ロバスト性を高めるための、データ水増し用。上2つと同様のディレクトリで行う。