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

Compare commits

base: Hazel:5baefdcc6ff85f3c9749043a07e2239476029e1e
Branches Tags
Hazel:main
...
compare: Hazel:8bd512a0a765168eb8d1779e07f0bd130225e0b8
Branches Tags
Hazel:main

3 Commits
5baefdcc6f ... 8bd512a0a7

Author SHA1 Message Date
Lars Noack
8bd512a0a7 feat: added alternative inpainting 2025-04-24 17:06:54 +02:00
Lars Noack
8fc56b887d feat: added alternative inpainting 2025-04-24 16:59:30 +02:00
Lars Noack
edad12841f fix: some minor errors 2025-04-24 16:55:44 +02:00
3 changed files with 8 additions and 5 deletions
Show Stats Expand all files Collapse all files

4
secure_pixelation/__main__.py
View File

@ -6,5 +6,5 @@ def cli():
print(f"Running secure_pixelation") print(f"Running secure_pixelation")
pixelate("assets/human_detection/test.png", generative_impaint=True) pixelate("assets/human_detection/test.png", generative_impaint=True)
# pixelate("assets/human_detection/humans.png", generative_impaint=False) pixelate("assets/human_detection/humans.png", generative_impaint=False)
# pixelate("assets/human_detection/rev1.png", generative_impaint=False) pixelate("assets/human_detection/rev1.png", generative_impaint=False)

5
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")
@ -84,9 +84,12 @@ def pixelate(to_detect: str, generative_impaint: bool = True, debug_drawings: bo
if generative_impaint: if generative_impaint:
step_2 = do_generative_impaint(raw_image, image=step_1) step_2 = do_generative_impaint(raw_image, image=step_1)
step_2_alt = quick_impaint(raw_image, image=step_1)
else: else:
step_2 = quick_impaint(raw_image, image=step_1) 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, "step_2")
write_image(step_2_alt, "step_2_alt")
step_3 = pixelate_regions(raw_image, image=step_2) step_3 = pixelate_regions(raw_image, image=step_2)
write_image(step_3, "step_3") write_image(step_3, "step_3")

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