From 1dd387d9807bc178bcd08a5d17b1278c93b944e6 Mon Sep 17 00:00:00 2001 From: Lars Noack Date: Wed, 23 Apr 2025 17:23:46 +0200 Subject: [PATCH] feat: improved stuff --- secure_pixelation/__main__.py | 2 +- secure_pixelation/detect_humans.py | 11 +++++++++++ 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/secure_pixelation/__main__.py b/secure_pixelation/__main__.py index b8aadd4..cc4b6cd 100644 --- a/secure_pixelation/__main__.py +++ b/secure_pixelation/__main__.py @@ -4,4 +4,4 @@ from .detect_humans import detect_humans def cli(): print(f"Running secure_pixelation") - detect_humans("assets/human_detection/rev1.png") + detect_humans("assets/human_detection/humans.png") diff --git a/secure_pixelation/detect_humans.py b/secure_pixelation/detect_humans.py index 16d9378..34b1f2b 100644 --- a/secure_pixelation/detect_humans.py +++ b/secure_pixelation/detect_humans.py @@ -241,6 +241,14 @@ def detect_human_parts(human: dict, face_padding: int = 20): for point in clean_points: cv2.circle(image, (int(point[0]), int(point[1])), 4, color, -1) + face_info = human["face"] = { + "is_valid": valid_face, + "x": face_box_x1, + "y": face_box_y1, + "w": face_box_x2 - face_box_x1, + "h": face_box_y2 - face_box_y1, + } + if valid_face: print("\nOriginal points:") print(original_points) @@ -364,3 +372,6 @@ def detect_humans(to_detect: str, crop_padding: int = 20, skip_detection_if_pres for human in human_boxes: detect_human_parts(human) + + with open(boxes_file, "w") as f: + json.dump(boxes_structures, f)