From a0657caf3e1685f687770d9d3fe48eb57ee4a450 Mon Sep 17 00:00:00 2001 From: Bingsu Date: Sat, 13 May 2023 15:00:42 +0900 Subject: [PATCH] fix: mask_preprocess values --- scripts/!adetailer.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py index 7db5352..d2f626d 100644 --- a/scripts/!adetailer.py +++ b/scripts/!adetailer.py @@ -667,7 +667,12 @@ class AfterDetailerScript(scripts.Script): with ChangeTorchLoad(): pred = predictor(ad_model, pp.image, args.ad_conf, **kwargs) - masks = mask_preprocess(pred.masks, args.ad_dilate_erode) + masks = mask_preprocess( + pred.masks, + kernel=args.ad_dilate_erode, + x_offset=args.ad_x_offset, + y_offset=args.ad_y_offset, + ) if not masks: print(