白黒画像をカラー画像に変換するネットワークを作ってみた(pix2pix) - Qiita

1. はじめに GAN(敵対的生成ネットワーク)を用いて、グレースケール画像に自動彩色をしていきたいと思います。 技術的には「pix2pix」と呼ぶらしいです。 このグレースケール画像が 自動で以下のように彩色できました!! ところどころ、おかしなところもあったり、うまくいかない画像もあったりしますが、結構自然な着色になっています。 ちなみに元画像を一番下の段だけ示すと、 こんな感じ。電車やベッドの色は違ったりしますが、全体的には同じような色合いで塗れている気がします。 2.ざっくりとした今回の学習のイメージ ざっくりとした今回の学習のイメージは以下のようになります。 GANなのでGenerator, Discriminatorの2つのネットワークを用いてます。 (1) (2) (3) (4) (5) (6) このようにGeneratorとDiscriminator、 2つのネットワークを