fix: mask_preprocess values

This commit is contained in:
Bingsu
2023-05-13 15:00:42 +09:00
parent 30cb65e451
commit a0657caf3e

View File

@@ -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(