HINet

Half Instance Normalization Network for Image Restoration, based on https://github.com/megvii-model/HINet.

Dependencies

NumPy
PyTorch, preferably with CUDA. Note that torchvision and torchaudio are not required and hence can be omitted from the command.
VapourSynth
(Optional) TensorRT. Note that uff and PyCUDA are not required and hence can be skipped from the guide.
(Optional) torch2trt

pip install --upgrade vshinet
python -m vshinet

from vshinet import HINet

ret = HINet(clip)

See __init__.py for the description of the parameters.