From 6eea2a911fc51f4317ebf19b2fe37083ba93f0d5 Mon Sep 17 00:00:00 2001
From: Won-Kyu Park <wkpark@gmail.com>
Date: Sat, 13 May 2023 14:59:25 +0900
Subject: [PATCH] fix: use dilate_erode factor correctly (#47)

---
 scripts/!adetailer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py
index 3058bf0..2f36004 100644
--- a/scripts/!adetailer.py
+++ b/scripts/!adetailer.py
@@ -611,7 +611,7 @@ class AfterDetailerScript(scripts.Script):
         with ChangeTorchLoad():
             pred = predictor(ad_model, pp.image, args.ad_conf, **kwargs)
 
-        masks = mask_preprocess(pred.masks)
+        masks = mask_preprocess(pred.masks, args.ad_dilate_erode)
 
         if not masks:
             print(