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

feat: fixed some issue

Browse Source
This commit is contained in:
Hazel 2025-04-23 17:14:17 +02:00
parent ad8f3b8e66
commit 0e73aff25d
2 changed files with 16 additions and 13 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/humans.png") detect_humans("assets/human_detection/rev1.png")

27
secure_pixelation/detect_humans.py
View File

@ -159,6 +159,9 @@ def detect_human_parts(human: dict, face_padding: int = 20):
# indices of the points that seem to be likely correct # indices of the points that seem to be likely correct
success_points = [] success_points = []
for i in range(5): for i in range(5):
if np.sum(original_points[i]) == 0:
continue
s_count = 0 s_count = 0
for j in range(5): for j in range(5):
d = np.abs(optimized_distances[i][j]) d = np.abs(optimized_distances[i][j])
@ -238,18 +241,18 @@ 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)
if valid_face:
print("\nOriginal points:") print("\nOriginal points:")
print(original_points) print(original_points)
print("\nOriginal pairwise distances:") print("\nOriginal pairwise distances:")
print(linearize_pairwise_distances(original_points, relative_face_matrix)) print(linearize_pairwise_distances(original_points, relative_face_matrix))
print(f"Optimized rotation vector (axis-angle): {rotation_vector}") print(f"Optimized rotation vector (axis-angle): {rotation_vector}")
print("\nOptimized points after rotation:") print("\nOptimized points after rotation:")
print(optimized_points) print(optimized_points)
print("\nOptimized pairwise distances:") print("\nOptimized pairwise distances:")
print(optimized_distances) print(optimized_distances)
print(success_points) print(success_points)
print(clean_points) print(clean_points)