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

feat: implemented effective but non generative impainting

Browse Source
This commit is contained in:
Hazel 2025-04-24 11:41:46 +02:00
parent 180b41ffa4
commit 678aeab7a5
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
secure_pixelation/pixelation_process.py
View File

@ -1,5 +1,6 @@
from __future__ import annotations
from typing import Optional
from pathlib import Path
import cv2
@ -17,6 +18,19 @@ def blackout(raw_image: RawImage) -> np.ndarray:
return image
def impaint(raw_image: RawImage, image: Optional[np.ndarray] = None) -> np.ndarray:
image = image if image is not None else raw_image.get_image()
# Create a mask where blacked-out areas are marked as 255 (white)
mask = np.zeros(image.shape[:2], dtype=np.uint8)
for (x, y, w, h) in raw_image.bounding_boxes:
mask[y:y+h, x:x+w] = 255
# Apply inpainting using the Telea method
return cv2.inpaint(image, mask, inpaintRadius=3, flags=cv2.INPAINT_TELEA)
def pixelate(to_detect: str, debug_drawings: bool = True):
raw_image = RawImage(to_detect)
@ -34,3 +48,6 @@ def pixelate(to_detect: str, debug_drawings: bool = True):
step_1 = blackout(raw_image)
write_image(step_1, "step_1")
step_2 = impaint(raw_image, image=step_1)
write_image(step_2, "step_2")