Metadata-Version: 2.3
Name: vsdpir
Version: 4.2.0
Summary: DPIR function for VapourSynth
Project-URL: Homepage, https://github.com/HolyWu/vs-dpir
Project-URL: Issues, https://github.com/HolyWu/vs-dpir/issues
Author-email: HolyWu <holywu@gmail.com>
Keywords: DPIR,PyTorch,TensorRT,VapourSynth
Classifier: Environment :: GPU :: NVIDIA CUDA
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3
Classifier: Topic :: Multimedia :: Video
Requires-Python: >=3.10
Requires-Dist: numpy
Requires-Dist: requests
Requires-Dist: torch>=2.5.1
Requires-Dist: tqdm
Requires-Dist: vapoursynth>=66
Description-Content-Type: text/markdown

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


## Dependencies
- [PyTorch](https://pytorch.org/get-started/) 2.5.1 or later
- [VapourSynth](http://www.vapoursynth.com/) R66 or later

`trt` requires additional packages:
- [TensorRT](https://developer.nvidia.com/tensorrt) 10.3.0 or later
- [Torch-TensorRT](https://pytorch.org/TensorRT/) 2.5.0 or later

To install the latest nightly build of PyTorch and Torch-TensorRT, run:
```
pip install -U packaging setuptools wheel
pip install --pre -U torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu124
pip install --no-deps --pre -U torch_tensorrt --index-url https://download.pytorch.org/whl/nightly/cu124
pip install -U tensorrt-cu12 tensorrt-cu12_bindings tensorrt-cu12_libs --extra-index-url https://pypi.nvidia.com
```


## Installation
```
pip install -U vsdpir
python -m vsdpir
```


## Usage
```python
from vsdpir import dpir

ret = dpir(clip)
```

See `__init__.py` for the description of the parameters.
