Real-ESRGAN を構築するための Ubuntu+AMD グラフィックス カード

現時点では、Ubuntu20.04+rocm5.2 でのみ正常にビルドされており、ビルドするには root 権限に切り替える必要があります。そうしないと、Real-ESRGAN が nvidia ドライバーを自動的にダウンロードします (環境変数に追加する必要がある場合があります)。 、リンクを構築します: Ubuntu22.04 ドライバーに AMD グラフィック カードをインストールし、pytorch_V2077 ブログ - CSDN ブログ、プロジェクト リンク: GitHub - xinntao/Real-ESRGAN: Real-ESRGAN は、一般的な画像/ビデオ復元のための実用的なアルゴリズムを開発することを目的としています。

 Real-ESRGAN をインストールする

$ cd ~
$ git clone https://github.com/xinntao/Real-ESRGAN.git
$ cd Real-ESRGAN

# 安装 basicsr - https://github.com/xinntao/BasicSR
# 我们使用BasicSR来训练以及推断
$ pip install basicsr
# facexlib和gfpgan是用来增强人脸的
$ pip install facexlib
$ pip install gfpgan
$ pip install -r requirements.txt
$ sudo python3 setup.py develop

 画像の修復

sudo python3 inference_realesrgan.py -n RealESRGAN_x4plus -i inputs --face_enhance

ビデオの修復

インストール

sudo apt-get -y install ffmpeg && pip3 install ffmpeg

 修理

# rmvb转mp4
$ ffmpeg -i [inputname].rmvb -c:v libx264 -strict -2 [outputname].mp4

$ sudo python3 inference_realesrgan_video.py -n RealESRGAN_x4plus -i inputs.mp4 --face_enhance

また

$ ffmpeg -i input.mp4 -qscale:v 1 -qmin 1 -qmax 1 -vsync 0 ./inputs/frame%08d.jpg
$ sudo python3 inference_realesrgan.py -n RealESRGAN_x4plus -i inputs --face_enhance
$ ffmpeg -i ./results/frame%08d_out.jpg -i input.mp4 -map 0:v:0 -map 1:a:0 -c:a copy -c:v libx264 -r 23.98 -pix_fmt yuv420p output.mp4

Python スクリプトの使用法 

使用法: python inference_realesrgan.py -n RealESRGAN_x4plus -i infile -o outfile [オプション]...

一般的なコマンド: python inference_realesrgan.py -n RealESRGAN_x4plus -i infile --outscale 3.5 --face_enhance

  -h このヘルプを表示
  -i --input 画像またはフォルダーを入力します。デフォルト: inputs
  -o --output 出力フォルダー。デフォルト: results
  -n --model_name モデル名。デフォルト: RealESRGAN_x4plus
  -s, --outscale 画像の最終的なアップサンプリング スケール。デフォルト: 4
  --suffix 復元されたイメージのサフィックス。デフォルト: out
  -t、--tile タイル サイズ。テスト中にタイルがない場合は 0。デフォルト: 0
  --face_enhance GFPGAN を使用して顔を強化するかどうか。デフォルト: False
  --fp32 推論中に fp32 精度を使用します。デフォルト: fp16 (半精度)。
  --ext 画像拡張子。オプション: 自動 | jpg | png、auto は入力と同じ拡張子を使用することを意味します。デフォルト: 自動

モデルのダウンロード:

  1. RealESRGAN_x4plus
  2. RealESRNet_x4plus
  3. RealESRGAN_x2plus
  4. realr-animevideov3
  5. RealESRGAN_x4plus_anime_6B

おすすめ

転載: blog.csdn.net/u013636118/article/details/129815508