Half Instance Normalization Network for Image Restoration, based on https://github.com/megvii-model/HINet.
- NumPy
- PyTorch, preferably with CUDA. Note that
torchvision
andtorchaudio
are not required and hence can be omitted from the command. - VapourSynth
- (Optional) TensorRT. Note that
uff
andPyCUDA
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.