U-GAT-ITを手元で動かす手順

Selfie 2 Waifuという写真をアニメ調に変換してくれるサイトがあって、

http://waifu.lofiu.com/

これがなかなかすごい。

Selfie 2 waifuhttps://t.co/Mm6vhQuQKDhttps://t.co/sjtHNM20Ln pic.twitter.com/4hEIoc4so7

— Shuichi Tsutsumi (@shu223) April 17, 2020

これはU-GAT-IT（Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation）という手法を用いていて、TensorFlowの実装が公開されている。

taki0112/UGATIT

ちなみにオリジナル（論文著者？）のPyTorch実装もある。

で、本記事はTF実装をgit cloneして手元で試してみた手順のメモ。環境としてはPython 3.7、TensorFlow 1.14。

出力の例：