参考記事
モノクロ画像をAIでカラー化してくれるPythonツールを試してみた
Mac上でモノクロ画像をカラー化するAIを動かしてみたいと思います。
colorization
Colorizationをgit cloneする
Colorizationというモデルを使います。
たくさんのソースが展開されていてほんと助かりますね。
git clone https://github.com/richzhang/colorization.git
cloneできたらcolorizationフォルダへ移動します。
cd colorization
ライブラリのインストール
requirements.txtに書いてあるライブラリが古いので書き換えます。
skimage → scikit-image
PIL → pillow
pip install -r requirements.txt
pytorchを使っているみたいですね。
カラー化に挑戦
お試しでサンプルの画像をカラー化してみます。
python demo_release.py -i ./imgs/ansel_adams.jpg
結構時間がかかってますねー。
Downloading: "https://colorizers.s3.us-east-2.amazonaws.com/colorization_release_v2-9b330a0b.pth" to /Users/***/.cache/torch/hub/checkpoints/colorization_release_v2-9b330a0b.pth
100%|███████████████████████████████████████████████████████████████████████████████| 123M/123M [01:40<00:00, 1.28MB/s]
Downloading: "https://colorizers.s3.us-east-2.amazonaws.com/siggraph17-df00044c.pth" to /Users/***/.cache/torch/hub/checkpoints/siggraph17-df00044c.pth
100%|███████████████████████████████████████████████████████████████████████████████| 130M/130M [01:45<00:00, 1.29MB/s]
モデルのダウンロードに時間がかかっただけみたい。
出力された画像がこちら
すごい！！
別のサンプルでやると瞬時にでるのでものすごく速いですね。
けどこれだと元々どのような画像だったのかわからないのでそれらしい色がついただけのように見えます。
オリジナルの画像でためす
つぎにオリジナル画像をやってみます。
こちらの呼子の写真を、モノクロ化します。
Macの写真アプリで簡単にできます。
できました。これをcolorizationしてみます。
空の色はいけてますが、イカさんのピンクはこの看板オリジナルなので、再現はできていませんね。
おもしろいですねー。
簡単にためせるのでぜひやってみてください！
古い写真とかやってみるとおもしろいかも！