人工知能で白黒写真をカラーにするプログラムを試してみました

人工知能で白黒写真をカラーに　早大が技術開発、GitHubでコード公開という記事を読みました。ソースが公開されていたので、試してみました。

ソース

• siggraph2016_colorization
• ディープネットワークを用いた大域特徴と局所特徴の学習による 白黒写真の自動色付け
• Getting started with Torch

MacOS X El Capitan‎のMacbook Airで検証します。ソースを見ると、Luaを使っているようです。

まず、Torchのインストール

$ git clone https://github.com/torch/distro.git ~/torch --recursive
$ cd ~/torch; bash install-deps;
$ ./install.sh
$ source ~/.profile
$ th

  ______             __   |  Torch7
 /_  __/__  ________/ /   |  Scientific computing for Lua.
  / / / _ \/ __/ __/ _ \  |  Type ? for help
 /_/  \___/_/  \__/_//_/  |  https://github.com/torch
                          |  http://torch.ch

th>

siggraph2016_colorizationをgit cloneして、必要なモデルをダウンロードします。

$ ./download_model.sh

下記のコマンドで白黒写真にカラーの色を付けます。

$ th colorize.lua ansel_colorado_1941.png out.png

いくつかの写真を変換してみました。

サンプルとして付いている山の写真。

外国のどこかの駅の写真

明治時代あたりと思われる写真

朝日か夕暮れ前の写真

私の子どもの頃のモノクロ写真

モノクロでも保存状態の良くないものは、綺麗な色は付かない感じがします。