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

feat: improved stuff

Browse Source
This commit is contained in:
Hazel 2025-04-23 17:23:46 +02:00
parent 9ae09b1ae5
commit 1dd387d980
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
secure_pixelation/__main__.py
View File

@ -4,4 +4,4 @@ from .detect_humans import detect_humans
def cli(): def cli():
print(f"Running secure_pixelation") print(f"Running secure_pixelation")
detect_humans("assets/human_detection/rev1.png") detect_humans("assets/human_detection/humans.png")

11
secure_pixelation/detect_humans.py
View File

@ -241,6 +241,14 @@ def detect_human_parts(human: dict, face_padding: int = 20):
for point in clean_points: for point in clean_points:
cv2.circle(image, (int(point[0]), int(point[1])), 4, color, -1) 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: if valid_face:
print("\nOriginal points:") print("\nOriginal points:")
print(original_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: for human in human_boxes:
detect_human_parts(human) detect_human_parts(human)
with open(boxes_file, "w") as f:
json.dump(boxes_structures, f)