Google Colaboratory及びDeOldifyによるモノクロ画像の自動色付け

python

Pythonで画像処理してみたいです。

このような要望にお応えします。

今回は、モノクロ画像に自然な色付けを自動的に行うことができるDeOldifyを利用します。

下記のサイトを参考にさせていただきました。

  • https://github.com/jantic/DeOldify
  • https://blog.shikoan.com/sagan-self-attention/
  • http://urusulambda.com/2018/07/15/saganself-attention-generative-adversarial-networkのself-attention機構をざっくり理解した/
  • https://qiita.com/shionhonda/items/330c9fdf78e62db3402b
  • https://gigazine.net/news/20181105-deoldify-colorize-image/

DeOldifyについて

DeOldifyは、SAGANモデルが用いられています。SAGAN(SelfAttentionGAN)については、以下の記事がよいと思いました。

  • http://urusulambda.com/2018/07/15/saganself-attention-generative-adversarial-networkのself-attention機構をざっくり理解した/
  • https://blog.shikoan.com/sagan-self-attention/
  • https://qiita.com/shionhonda/items/330c9fdf78e62db3402b

今回は、下記で公開されているライブラリを用いてモノクロ画像の自動色付けをします。
GitHub – jantic/DeOldify: A Deep Learning based project for colorizing and restoring old images

  • https://github.com/jantic/DeOldify

Google Colaboratoryの準備

・Googleのアカウントを作成します。
・Googleドライブにアクセスし、「新規」→「その他」から「Google Colaboratory」の順でクリックします。そうすると、Colaboratoryが起動します。

・Colaboratoryが起動したら、以下のコマンドをCoalboratoryのセルに入力し実行します。
そうすることで、Googleドライブをマウントします。

・実行後、認証コードの入力が促されます。このとき、「Go to this URL in a browser」が指しているURLにアクセスしgoogleアカウントを選択すると、認証コードが表示されますので、それをコピーしenterを押します。これでGoogleドライブのマウントが完了します。

DeOldifyの準備

Google Colaboratoryの「ランタイム」→「ランタイムのタイプ変更」でGPUを選択します。

ツールをダウンロードする場所に移動します。
本記事では、マイドライブにダウンロードします。

gitからツールをダウンロードします。

ダウンロードしたDeOldifyまで移動します。

GPUの確認をします。

ライブラリ実行に必要なライブラリをインストールします。

ツール実行に必要なライブラリをimportします。

modelsフォルダを作成し、モデルデータを取得します。

サンプル画像を取得します。

モノクロ画像に色付けします。

出力結果

ダウンロードしたDeOldifyにあるtest_imagesに入力するモノクロ画像を格納し実行しました。

どうでしょうか?
今回は、DeOldifを用いてモノクロ画像の自動色付けを行ってみました。より複雑なモノクロ画像に対して実施してみるのもいいかもしれません。

タイトルとURLをコピーしました