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")