Ubuntu22.04DesktopでReal-ESRGANのインストール記録

Real-ESRGANをインストールしたのでその作業ログを残しておきます。

pythonの仮想環境を作成

cd ~/git 
python3 -m venv resrganvenv

環境名をresrganvenvにしました。

仮想環境の開始

source ~/git/resrganvenv/bin/activate

仮想環境の終了要確認

deactivate

必要なパッケージをインストール

python -m pip install --upgrade pip
python -m pip install -U --ignore-installed pip
python -m pip install -U torch torchvision torchaudio numpy --index-url https://download.pytorch.org/whl/cu115
cd ~/git
git clone --recursive https://github.com/xinntao/Real-ESRGAN
cd Real-ESRGAN
vi requirements.txt

以下のように編集

basicsr
facexlib
gfpgan
numpy
opencv-python
Pillow
torch
torchvision
tqdm

python -m pip install -r requirements.txt
python setup.py develop

インストール終了

動作確認

python inference_realesrgan.py -n RealESRGAN_x4plus -i inputs -o results  --outscale 4 --suffix 4x

~/git/Real-ESRGAN/resultsに画像が出来上がっていると成功（videoディレクトリでコケるがOK)

仮想環境の開始と入出力ディレクトリを指定した実行用のシェルスクリプト

#!/bin/bash
cd ~
venvFile=`pwd`'/git/resrganvenv/bin/activate'
dataRoot='/mnt/workspace'
inputDir=$dataRoot'/inputs'
outputDir=$dataRoot'/outputs'
logfile=$dataRoot'/logs/Real-ESRGAN.log'

cd $outputDir
if [ `ls -U1 | wc -l` -ne 0 ]; then
    rm *
fi;
cd $inputDir
if [ `ls -U1 | wc -l` -eq 0 ]; then
    echo 'Not File';
    exit 1
fi;
cd ~/git/Real-ESRGAN
source $venvFile
echo '*** Real-ESRGAN Start ***' >> $logfile
date >> $logfile
python inference_realesrgan.py -n RealESRGAN_x4plus --tile 1024 -i $inputDir -o $outputDir >> $logfile
date >> $logfile
echo '*** Real-ESRGAN End ***' >> $logfile
deactivate
cd $inputDir
if [ `ls -U1 | wc -l` -ne 0 ]; then
    rm *
fi;

実行環境に合わせて修正する必要あり。
自分の環境では画像サイズが1024×1024ぐらいまでは処理できるがそれ以上の場合メモリ不足になるので、tileオプションをつけています。