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 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) raw_image = RawImage(to_detect)
step_dir = raw_image.get_dir("steps") step_dir = raw_image.get_dir("steps")

4
secure_pixelation/simple_lama_bindings.py
View File

@ -1,5 +1,5 @@
import os import os
from typing import Optional from typing import Tuple
import torch import torch
import cv2 import cv2
@ -11,7 +11,7 @@ from huggingface_hub import hf_hub_download
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1" 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): def get_image(img: np.ndarray):
img = img.copy() img = img.copy()