add warning on improved mask and upscaling and make it disabled in settings by default

main
Tran Xen 2 years ago
parent db79243acc
commit e6592a11bf

@ -201,7 +201,7 @@ def on_ui_settings() -> None:
shared.opts.add_option( shared.opts.add_option(
"faceswaplab_default_upscaled_swapper_improved_mask", "faceswaplab_default_upscaled_swapper_improved_mask",
shared.OptionInfo( shared.OptionInfo(
True, False,
"Default Use improved segmented mask (use pastenet to mask only the face) (requires restart)", "Default Use improved segmented mask (use pastenet to mask only the face) (requires restart)",
gr.Checkbox, gr.Checkbox,
{"interactive": True}, {"interactive": True},

@ -210,6 +210,11 @@ class UpscaledINSwapper(INSwapper):
) )
if options.improved_mask: if options.improved_mask:
if k == 1:
logger.warning(
"Please note that improved mask does not work well without upscaling. Set upscaling to Lanczos at least if you want speed and want to use improved mask."
)
logger.info("improved_mask") logger.info("improved_mask")
mask = get_face_mask(aimg, bgr_fake) mask = get_face_mask(aimg, bgr_fake)
bgr_fake = merge_images_with_mask(aimg, bgr_fake, mask) bgr_fake = merge_images_with_mask(aimg, bgr_fake, mask)

Loading…
Cancel
Save