写真解像度を上げる（高画質化）Pythonツール「SwinIR」を試してみた。

写真解像度を上げる「SwinIR」を試してみた。

「SwinIR」を利用するためには、「Python」が必要になるので、Pythonをインストールしておく。

【SwinIR 導入手順】

■ 「Python」パッケージライブラリ「opencv-python」のインストール

Windowsコマンドプロンプトで、次のコマンドを入れる。「opencv-python」以外のものも一緒にインストール。（pip は、Pythonと一緒にインストールされるコマンド）

pip install opencv-python tqdm timm ipython

■ Githubから「SwinIR」をダウンロード

https://github.com/JingyunLiang/SwinIR

[Code] ＞ [Download ZIP]で、ダウンロードしたものを解凍して任意のフォルダーに入れる。今回、以下のフォルダーにした。

D:\MyData\SwinIR-main

■ 学習済みモデルのダウンロード

https://github.com/JingyunLiang/SwinIR/releases/

から、以下の2つをダウンロードして試してみた。

003_realSR_BSRGAN_DFOWMFC_s64w8_SwinIR-L_x4_GAN.pth
003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x2_GAN.pth

「003_realSR_BSRGAN_DFOWMFC_s64w8_SwinIR-L_x4_GAN.pth」は、解像度を4倍にしてくれる。2倍にならないか試したがうまくいかなかった。

「003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x2_GAN.pth」は、解像度を2倍にしてくれる。

フォルダー「D:\MyData\SwinIR-main」の下に「003_realSR_BSRGAN_DFOWMFC_s64w8_SwinIR-L_x4_GAN.pth」「003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x2_GAN.pth」を入れる。

■ 実行

元画像を以下のフォルダーに入れて、下のコマンドを入力すると、

D:\MyData\SwinIR-main\testsets\sample

以下のフォルダーの下に「swinir_real_sr_x4_large」「swinir_real_sr_x2」等のサブフォルダーが出来て、結果の画像が入る。

D:\MyData\SwinIR-main\results

実行コマンドは以下。

≪「003_realSR_BSRGAN_DFOWMFC_s64w8_SwinIR-L_x4_GAN.pth」の場合≫

d:
cd D:\MyData\SwinIR-main
python main_test_swinir.py --task real_sr --model_path 003_realSR_BSRGAN_DFOWMFC_s64w8_SwinIR-L_x4_GAN.pth --folder_lq testsets/sample --scale 4 --large_model

≪「003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x2_GAN.pth」の場合≫

d:
cd D:\MyData\SwinIR-main
python main_test_swinir.py --task real_sr --model_path 003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x2_GAN.pth --folder_lq testsets/sample --scale 2

処理時間は結構かかる。PCの能力や画像のサイズにもよるが10分程度はかかる。