From 6f93643159c6cc683e09547be9985274e13f2a38 Mon Sep 17 00:00:00 2001 From: mawen1250 Date: Tue, 26 Jan 2016 19:24:49 +0800 Subject: [PATCH] BM3D: fixed format checking for "pre" and "flt" --- mvsfunc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mvsfunc.py b/mvsfunc.py index bb9a593..d948b95 100644 --- a/mvsfunc.py +++ b/mvsfunc.py @@ -878,7 +878,7 @@ def BM3D(input, sigma=None, radius1=None, radius2=None, profile1=None, profile2= if pre is not None: if not isinstance(pre, vs.VideoNode): raise TypeError(funcName + ': \"pre\" must be a clip!') - if pre.format != sFormat: + if pre.format.id != sFormat.id: raise ValueError(funcName + ': clip \"pre\" must be of the same format as the input clip!') if pre.width != input.width or pre.height != input.height: raise ValueError(funcName + ': clip \"pre\" must be of the same size as the input clip!') @@ -886,7 +886,7 @@ def BM3D(input, sigma=None, radius1=None, radius2=None, profile1=None, profile2= if ref is not None: if not isinstance(ref, vs.VideoNode): raise TypeError(funcName + ': \"ref\" must be a clip!') - if ref.format != sFormat: + if ref.format.id != sFormat.id: raise ValueError(funcName + ': clip \"ref\" must be of the same format as the input clip!') if ref.width != input.width or ref.height != input.height: raise ValueError(funcName + ': clip \"ref\" must be of the same size as the input clip!')