画像をきれいにアップスケール
SwinIRを使います。
元画像
結果(4倍スケール)
使い方
・SwinIRのリポジトリをクローン、
・事前トレーニング済みモデルをダウンロード、
・画像フォルダを指定して実行します。
git clone https://github.com/JingyunLiang/SwinIR.git
cd SwinIR
pip install timm
wget https://github.com/JingyunLiang/SwinIR/releases/download/v0.0/003_realSR_BSRGAN_DFOWMFC_s64w8_SwinIR-L_x4_GAN.pth -P model_zoo
python main_test_swinir.py --task real_sr --scale 4 --large_model --model_path model_zoo/003_realSR_BSRGAN_DFOWMFC_s64w8_SwinIR-L_x4_GAN.pth --folder_lq path_to_your_image_directory
たくさんのトレーニング済みモデルが提供されている
ノイズ除去など、さまざまなデータセットでトレーニングされたSwinIRモデルが提供されており、好きなモデルを使えます。
🐣
フリーランスエンジニアです。
お仕事のご相談こちらまで
rockyshikoku@gmail.com
Core MLやARKitを使ったアプリを作っています。
機械学習/AR関連の情報を発信しています。