search

tongyuantongyu / vs-NNVISR

Neural Network Video Interpolation / Super Resolution Filter for VapourSynth
BSD 3-Clause "New" or "Revised" License
46 stars 1 forks source link

model path is wrong 'yuv_1_1_1' vs ' 'yuv_5_5_1' #3

Closed Selur closed 1 year ago

Selur commented 1 year ago

https://github.com/tongyuantongyu/vs-NNVISR/blob/main/docs/models.md shows 'yuv_1_1_1' as folder and extracting the models this folder is created, but when using (on a YUV420P8 source):

model_config = {'scale_factor': 2, 'input_count': 2, 'feature_count': 64, 'extraction_layers': 4, 'interpolation': True, 'extra_frame': True, 'double_frame': True, 'model': 'cycmunet/vimeo90k-deblur'}
clip = core.nnvisr.Super(clip, model_path="F:/Hybrid/64bit/vsfilters/ResizeFilter/NNVISR", **model_config)

I get:

Source model file not exist:"F:/Hybrid/64bit/vsfilters/ResizeFilter/NNVISR\\models\\cycmunet/vimeo90k-deblur\\yuv_5_5_1\\fe_n2_2x2_l4_yuv1-1.onnx"

renaming the 'yuv_1_1_1'-folder to ' 'yuv_5_5_1' does seem to fix the issue.

Selur commented 1 year ago

Okay, the assumed model path is depening on the _Transfer, _Primaries and matrix Info in Vapoursynth,.. So Matrix, Transfer and Primaries all have to be bt709.