DPIR

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

Dependencies

• ncnn. Note that the official Python package is not built with Vulkan compute support. Windows users can install the wheel from https://github.com/HolyWu/ncnn/releases/latest. Non-windows users have to build from source using my fork (requires Vulkan SDK):

git clone https://github.com/HolyWu/ncnn.git --depth 1
cd ncnn
git submodule update --init --recursive --depth 1
pip install .

• VapourSynth R55 or later.

Installation

pip install -U vsdpir_ncnn
python -m vsdpir_ncnn

Usage

from vsdpir_ncnn import DPIR

dpir = DPIR(task='deblock')
ret = dpir.run(clip, strength=25)

See __init__.py for the description of the parameters.