From 8fc56b887dc7c22af02d1c029cbb92971c36e4ed Mon Sep 17 00:00:00 2001 From: Lars Noack Date: Thu, 24 Apr 2025 16:59:30 +0200 Subject: [PATCH] feat: added alternative inpainting --- secure_pixelation/pixelation_process.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/secure_pixelation/pixelation_process.py b/secure_pixelation/pixelation_process.py index 3bb17dd..574dcc7 100644 --- a/secure_pixelation/pixelation_process.py +++ b/secure_pixelation/pixelation_process.py @@ -84,9 +84,12 @@ def pixelate(to_detect: str, generative_impaint: bool = True, debug_drawings: bo if generative_impaint: step_2 = do_generative_impaint(raw_image, image=step_1) + step_2_alt = quick_impaint(raw_image, image=step_1) else: 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_alt, "step_2_alt") step_3 = pixelate_regions(raw_image, image=step_2) write_image(step_3, "step_3")