高解像度画像の画質を上げたい
画像レタッチ機械学習モデルは、計算リソースの制限があるため、高解像度の画像を補正するのが従来むずかしかった。。。
高画質対応モデル 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関連の情報を発信しています。