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

feat: added proper pixelation

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

21
secure_pixelation/pixelation_process.py
View File

@ -31,6 +31,22 @@ def impaint(raw_image: RawImage, image: Optional[np.ndarray] = None) -> np.ndarr
return cv2.inpaint(image, mask, inpaintRadius=3, flags=cv2.INPAINT_TELEA)
def pixelate_regions(raw_image: RawImage, image: Optional[np.ndarray] = None, pixel_size: int = 10) -> np.ndarray:
image = image.copy() if image is not None else raw_image.get_image().copy()
for (x, y, w, h) in raw_image.bounding_boxes:
roi = image[y:y+h, x:x+w]
# Resize down and then back up
temp = cv2.resize(roi, (max(1, w // pixel_size), max(1, h // pixel_size)), interpolation=cv2.INTER_LINEAR)
pixelated = cv2.resize(temp, (w, h), interpolation=cv2.INTER_NEAREST)
image[y:y+h, x:x+w] = pixelated
return image
def pixelate(to_detect: str, debug_drawings: bool = True):
raw_image = RawImage(to_detect)
@ -46,8 +62,13 @@ def pixelate(to_detect: str, debug_drawings: bool = True):
cv2.imwrite(f, image)
write_image(raw_image.image, "step_0")
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")
step_3 = pixelate_regions(raw_image, image=step_2)
write_image(step_3, "step_3")