CalvinBackup/sd-webui-faceswaplab
1
0
Fork 0
Code Issues 56 Pull Requests 4 Actions Packages Projects Releases 8 Wiki Activity

fix bugs, fix api, clear code

Browse Source
This commit is contained in:
Tran Xen
2023-07-29 00:24:51 +02:00
parent a511214aaa
commit d755be1325
11 changed files with 450 additions and 254 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/faceswaplab_postprocessing/postprocessing.py
+21 -6
View File
@@ -11,17 +11,32 @@ from scripts.faceswaplab_postprocessing.upscaling import upscale_img, restore_fa
def enhance_image(image: Image.Image, pp_options: PostProcessingOptions) -> Image.Image:
result_image = image
try:
if pp_options.inpainting_when == InpaintingWhen.BEFORE_UPSCALING.value:
result_image = img2img_diffusion(image, pp_options)
logger.debug("enhance_image, inpainting : %s", pp_options.inpainting_when)
result_image = image
if (
pp_options.inpainting_when == InpaintingWhen.BEFORE_UPSCALING.value
or pp_options.inpainting_when == InpaintingWhen.BEFORE_UPSCALING
):
logger.debug("Inpaint before upscale")
result_image = img2img_diffusion(result_image, pp_options)
result_image = upscale_img(result_image, pp_options)
if pp_options.inpainting_when == InpaintingWhen.BEFORE_RESTORE_FACE.value:
result_image = img2img_diffusion(image, pp_options)
if (
pp_options.inpainting_when == InpaintingWhen.BEFORE_RESTORE_FACE.value
or pp_options.inpainting_when == InpaintingWhen.BEFORE_RESTORE_FACE
):
logger.debug("Inpaint before restore")
result_image = img2img_diffusion(result_image, pp_options)
result_image = restore_face(result_image, pp_options)
if pp_options.inpainting_when == InpaintingWhen.AFTER_ALL.value:
result_image = img2img_diffusion(image, pp_options)
if (
pp_options.inpainting_when == InpaintingWhen.AFTER_ALL.value
or pp_options.inpainting_when == InpaintingWhen.AFTER_ALL
):
logger.debug("Inpaint after all")
result_image = img2img_diffusion(result_image, pp_options)
except Exception as e:
logger.error("Failed to upscale %s", e)