DPIR

Plug-and-Play Image Restoration with Deep Denoiser Prior, based on https://github.com/cszn/DPIR.

Dependencies

• NumPy
• ONNX Runtime. CUDA and TensorRT require onnxruntime-gpu, while DirectML requires onnxruntime-directml. Note that only one of onnxruntime, onnxruntime-gpu and onnxruntime-directml should be installed at a time in any one environment.
• VapourSynth R55 or newer.
• (Optional) CUDA Toolkit
• (Optional) cuDNN
• (Optional) TensorRT

Installation

pip install --upgrade vsdpir
python -m vsdpir

Usage

from vsdpir import DPIR

ret = DPIR(clip)

See __init__.py for the description of the parameters.