LoginSignup
29
29

More than 5 years have passed since last update.

白黒写真の自動色付けをDockerで試す

Posted at

早稲田大学 石川博教授チームが発表した自動色付け技術を自分の環境で試してみたいと思いました。

TL;DR 簡単な結論

$ git clone https://github.com/fuji70/auto_colorize.git
$ cd auto_colorize
$ ./run.sh input.png out.png

元論文のライセンス条項(非商用利用など)は守ってご利用下さい。

やったこと

著者自身による論文紹介から公開コードGithubが辿れます。
TorchのDockerイメージ kaixhin/torch を利用することで簡単に環境構築できました。
また変換実行は Docker 起動パラメータに入れてシェルスクリプトから呼び出すようにしました。

よって、前もってDocker環境(docker run hello-world が動く) をお持ちであれば下記に公開したスクリプトで簡単に自動彩色を試せます。

参考記事

29
29
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
29
29