グレイスケール画像をフルカラーにしてみた

深層学習用の開発ツール「Tourch」とカラー化ツールを使用することで、グレイスケールをフルカラー化できます。白黒とカラーの2種類の写真を組み合わせて学習させたモデルを使用しており、被写体の特徴から色を判断しています。ここでは、グレイスケール画像のフルカラー化について解説していきます。

01: 環境

  • Windows 7 Professional
  • VirtualBox 5.1.22
  • Vagrant 1.9.4
  • Vagrant: Ubuntu 14.04.5 LTS

02: インストール

bash


# gitのインストール
$ sudo apt-get update
$ sudo apt-get install git

# Torchのインストール
$ cd
$ git clone https://github.com/torch/distro.git ~/torch --recursive
$ cd torch
$ ./install-deps
...
Do you want to automatically prepend the Torch install location
to PATH and LD_LIBRARY_PATH in your /home/vagrant/.bashrc? (yes/no)
[yes] >>> 
yes
...
$ ./install.sh
$ source ~/.bashrc

# カラー化ツールのダウンロード
$ cd
$ git clone https://github.com/satoshiiizuka/siggraph2016_colorization.git
$ cd siggraph2016_colorization
$ ./download_model.sh

03: 動作テスト

bash


$ th colorize.lua ~/shared/grayscale.jpg ~/shared/color.jpg

grayscale.jpg

color.jpg