fix: use dilate_erode factor correctly (#47)

This commit is contained in:
Won-Kyu Park
2023-05-13 14:59:25 +09:00
committed by GitHub
parent 25d275fc4b
commit 6eea2a911f

View File

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