Hazel/secure_pixelation
1
0
Fork 0
generated from Hazel/python-project
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: some minor errors

Browse Source
This commit is contained in:
Hazel 2025-04-24 16:55:44 +02:00
parent 5baefdcc6f
commit edad12841f
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
secure_pixelation/pixelation_process.py
View File

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

4
secure_pixelation/simple_lama_bindings.py
View File

@ -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()