Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-04-30 19:21:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: mask_preprocess values

Browse Source
This commit is contained in:
Bingsu
2023-05-13 15:00:42 +09:00
parent 30cb65e451
commit a0657caf3e
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/!adetailer.py
View File

@@ -667,7 +667,12 @@ class AfterDetailerScript(scripts.Script):
with ChangeTorchLoad(): with ChangeTorchLoad():
pred = predictor(ad_model, pp.image, args.ad_conf, **kwargs) 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: if not masks:
print( print(