DPIR function for VapourSynth
Project description
DPIR
Plug-and-Play Image Restoration with Deep Denoiser Prior, based on https://github.com/cszn/DPIR.
Dependencies
- PyTorch 2.4.0.dev or later
- VapourSynth R66 or later
trt
requires additional Python packages:
- TensorRT 10.0.1
- Torch-TensorRT 2.4.0.dev
To install TensorRT, run pip install tensorrt==10.0.1 tensorrt-cu12_bindings==10.0.1 tensorrt-cu12_libs==10.0.1 --extra-index-url https://pypi.nvidia.com
To install Torch-TensorRT, Windows users can pip install the whl file on Releases. Linux users can run pip install --pre torch-tensorrt --index-url https://download.pytorch.org/whl/nightly/cu121
(requires PyTorch nightly build).
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.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
vsdpir-4.1.0-py3-none-any.whl
(13.3 kB
view hashes)