1580158
/
adetailer
mirror of https://github.com/Bing-su/adetailer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: mask_preprocess values

pull/57/head
Bingsu 2023-05-13 15:00:42 +09:00
parent 30cb65e451
commit a0657caf3e
1 changed files with 6 additions and 1 deletions
Show Stats 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(