CycleGANを用いたスタイル変換の続きです。
Cycle GANを使ってチーズケーキの画像⇔チョコレートケーキの画像ということをやりたかったのですが、前回記事を見てもらえばわかるように、あんまり上手くいっていませんでした。
結論だけ先に言ってしまうと、Normalization を Batch Normalization から Instance Normalization に変えたらいい感じの変換ができるようになりました。GPU のメモリの都合上、ミニバッチサイズが4 (チーズケーキとチョコレートケーキの画像を4枚ずつ) のような形になっていたので、流石にバッチサイズが小さすぎたのでしょう。
風景写真⇔MtGのイラストについても前よりは雰囲気出てます。
使ったコードはこちらに置いておきました。
今回はPyTorchでの実装です。
Normalization についても勉強して何か書きたいですね。
(2019年3月18日追記)
Pytorch 版で BatchNormalization 使った場合の結果について載せておくのを忘れたのでちょっと追記。
こんな感じなのでやっぱりあまり上手くいってないですね。