feat: added alternative inpainting

This commit is contained in:
Hazel 2025-04-24 16:59:30 +02:00
parent edad12841f
commit 8fc56b887d

View File

@ -84,9 +84,12 @@ def pixelate(to_detect: str, generative_impaint: bool = True, debug_drawings: bo
if generative_impaint: if generative_impaint:
step_2 = do_generative_impaint(raw_image, image=step_1) step_2 = do_generative_impaint(raw_image, image=step_1)
step_2_alt = quick_impaint(raw_image, image=step_1)
else: else:
step_2 = quick_impaint(raw_image, image=step_1) step_2 = quick_impaint(raw_image, image=step_1)
step_2_alt = do_generative_impaint(raw_image, image=step_1)
write_image(step_2, "step_2") write_image(step_2, "step_2")
write_image(step_2_alt, "step_2_alt")
step_3 = pixelate_regions(raw_image, image=step_2) step_3 = pixelate_regions(raw_image, image=step_2)
write_image(step_3, "step_3") write_image(step_3, "step_3")