Fix flat_mask thr scaling

Jaded-Encoding-Thaumaturgy · Nov 12, 2024 · cef1693 · cef1693
1 parent fb7ab11
commit cef1693
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/vsmasktools/spat_funcs.py b/vsmasktools/spat_funcs.py
@@ -149,7 +149,9 @@ def flat_mask(src: vs.VideoNode, radius: int = 5, thr: float = 0.011, gauss: boo
 
     blur = gauss_blur(luma, radius * 0.361083333) if gauss else box_blur(luma, radius)
 
-    mask = depth(luma, 8).abrz.AdaptiveBinarize(depth(blur, 8), scale_value(thr, 32, 8))
+    blur, mask = depth(blur, 8), depth(luma, 8)
+
+    mask = mask.abrz.AdaptiveBinarize(blur, scale_value(thr, 32, blur))
 
     return depth(mask, luma, dither_type=DitherType.NONE, range_in=ColorRange.FULL, range_out=ColorRange.FULL)