feat: live preview (#352)

This commit is contained in:
Won-Kyu Park
2023-10-03 20:03:37 +09:00
committed by GitHub
parent 910bf3b970
commit 46e4d94a46

View File

@@ -579,6 +579,7 @@ class AfterDetailerScript(scripts.Script):
pred = predictor(ad_model, pp.image, args.ad_confidence, **kwargs)
masks = self.pred_preprocessing(pred, args)
shared.state.assign_current_image(pred.preview)
if not masks:
print(