HINet
Half Instance Normalization Network for Image Restoration, based on https://github.com/megvii-model/HINet.
Dependencies
- NumPy
- PyTorch, preferably with CUDA. Note that
torchvisionandtorchaudioare not required and hence can be omitted from the command. - VapourSynth
- (Optional) TensorRT. Note that
uffandPyCUDAare not required and hence can be skipped from the guide. - (Optional) torch2trt
Installation
pip install --upgrade vshinet
python -m vshinet
Usage
from vshinet import HINet
ret = HINet(clip)
See __init__.py for the description of the parameters.
uploaded the full script and the source file to my