DPIR

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

Dependencies

• NumPy
• PyTorch >=2.0.1
• VapourSynth >=R60

trt requires additional runtime libraries:

• CUDA Toolkit 11.8
• cuDNN 8.9
• TensorRT >=8.6.1.6

For ease of installation on Windows, you can download the 7z file on Releases which contains required runtime libraries and Python wheel file. Either add the unzipped directory to your system PATH or copy the DLL files to a directory which is already in your system PATH. Finally pip install the Python wheel file.

Installation

pip install -U vsdpir
python -m vsdpir

Usage

from vsdpir import dpir

ret = dpir(clip)

See __init__.py for the description of the parameters.