diff --git a/vsaa/funcs.py b/vsaa/funcs.py index 8c1dca9..bf0cae5 100644 --- a/vsaa/funcs.py +++ b/vsaa/funcs.py @@ -9,8 +9,8 @@ from vsmasktools import EdgeDetect, EdgeDetectT, Prewitt, ScharrTCanny from vsrgtools import MeanMode, RepairMode, bilateral, box_blur, contrasharpening_median, repair, unsharp_masked from vstools import ( - MISSING, CustomRuntimeError, CustomValueError, FormatsMismatchError, FunctionUtil, KwargsT, MissingT, PlanesT, - VSFunction, get_h, get_peak_value, get_w, get_y, join, normalize_planes, plane, scale_8bit, scale_value, split, vs + MISSING, ColorRange, CustomRuntimeError, CustomValueError, FormatsMismatchError, FunctionUtil, KwargsT, MissingT, + PlanesT, VSFunction, get_h, get_peak_value, get_w, get_y, join, normalize_planes, plane, scale_value, split, vs ) from .abstract import Antialiaser, SingleRater @@ -423,7 +423,7 @@ def based_aa( if mask and not isinstance(mask, vs.VideoNode): mask = EdgeDetect.ensure_obj(mask, based_aa) mask = mask.edgemask(plane(func.work_clip, 0)) - mask = mask.std.Binarize(scale_8bit(func.work_clip, min(mask_thr, 255))) + mask = mask.std.Binarize(scale_value(min(mask_thr, 255), 8, func.work_clip, ColorRange.FULL)) mask = box_blur(mask.std.Maximum()).std.Limiter() if show_mask: