高解像度画像の画質を上げたい

画像レタッチ機械学習モデルは、計算リソースの制限があるため、高解像度の画像を補正するのが従来むずかしかった。。。

高画質対応モデル　Image-Adaptive-LUT

画像に適応したLookUpTableを学習し、しかも高画質、高速の処理をできるモデルです。

3345*4480 の画像でも、ColabのGPUで５秒で処理できました。

input / output

使用法

リポジトリをクローン。
必要なモジュールをインストールし、セットアップします。
少し古いバージョンのtorchなどをインストールする必要があります。
requirementsのPILとOpenCV-Pythonは最新バージョンを指定しなおします。

pip install torch==0.4.1.post2 torchvision===0.2.2 -f https://download.pytorch.org/whl/torch_stable.html
pip install -r requirements
cd trilinear_c
sh make.sh

実行。

python3 demo_eval.py --image_dir demo_images --image_name my_image.jpg --input_color_space sRGB

🐣

フリーランスエンジニアです。
お仕事のご相談こちらまで
rockyshikoku@gmail.com

Core MLやARKitを使ったアプリを作っています。
機械学習／AR関連の情報を発信しています。

Twitter
Medium
GitHub