Real ESRGAN-TPU

Real-ESRGAN-TPU 视频/图片分辨率超分应用，通过使用腾讯 ARC 实验室开源 Real-ESRGAN 系列模型，使用 Sophon SDK 移植到 SG2300X 芯片系列产品上，实现本地 TPU 进行硬件加速推理， 此应用可用于对任何低分辨率的视频/图片进行超分，最大分辨率可达 2560x1920，并提供多种针对不同场景使用的优化模型，可对真实场景，动漫视频进行针对性优化。 加入真实人面增强 CodeFormer 与 GFPGAN 模型。 加入 RMBG 针对主体进行背景移除模型，并可 DIY 任意背景（如证件照背景）。 此应用使用 Gradio 实现用户友好交互界面。

应用安装

• 克隆仓库

  git clone https://github.com/zifeng-radxa/Real-ESRGAN-tpu_Web.git

• 克隆人脸增强工具箱

  cd Real-ESRGAN-tpu_Web
  git clone https://github.com/zifeng-radxa/FACEXLIB.git

• 创建虚拟环境

  必须创建虚拟环境，否则可能会影响其他应用的正常运行， 虚拟环境使用请参考这里

  python3 -m virtualenv .venv
  source .venv/bin/activate

• 下载依赖

  pip3 install --upgrade pip setuptools wheel
  pip3 install -r requirements.txt
  pip3 install https://github.com/radxa-edge/TPU-Edge-AI/releases/download/v0.1.0/tpu_perf-1.2.31-py3-none-manylinux2014_aarch64.whl

• 下载模型

  python3 download_models.py

• 启动应用

  • 通过 gradio 启动

  bash run_gr.sh

  浏览器访问 Airbox ip 地址的 7860 端口

应用展示