diff --git a/secure_pixelation/pixelation_process.py b/secure_pixelation/pixelation_process.py index 8f38394..3bb17dd 100644 --- a/secure_pixelation/pixelation_process.py +++ b/secure_pixelation/pixelation_process.py @@ -62,7 +62,7 @@ def pixelate_regions(raw_image: RawImage, image: Optional[np.ndarray] = None, pi return image -def pixelate(to_detect: str, generative_impaint: bool = True, debug_drawings: bool = True): +def pixelate(to_detect: str, generative_impaint: bool = True, debug_drawings: bool = False): raw_image = RawImage(to_detect) step_dir = raw_image.get_dir("steps") diff --git a/secure_pixelation/simple_lama_bindings.py b/secure_pixelation/simple_lama_bindings.py index 654aa12..bb3ecd9 100644 --- a/secure_pixelation/simple_lama_bindings.py +++ b/secure_pixelation/simple_lama_bindings.py @@ -1,5 +1,5 @@ import os -from typing import Optional +from typing import Tuple import torch import cv2 @@ -11,7 +11,7 @@ from huggingface_hub import hf_hub_download os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1" -def prepare_img_and_mask(image: np.ndarray, mask: np.ndarray, device: torch.device, pad_out_to_modulo: int = 8, scale_factor: float = 1) -> tuple[torch.Tensor, torch.Tensor]: +def prepare_img_and_mask(image: np.ndarray, mask: np.ndarray, device: torch.device, pad_out_to_modulo: int = 8, scale_factor: float = 1) -> Tuple[torch.Tensor, torch.Tensor]: def get_image(img: np.ndarray): img = img.copy()