Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -416,7 +416,6 @@ holistic = mp_holistic.Holistic(static_image_mode=True, min_detection_confidence
|
|
416 |
|
417 |
# Lists of landmark names for pose, face, and hands
|
418 |
pose_landmarks = [name for name in mp_holistic.PoseLandmark.__members__]
|
419 |
-
face_landmarks = [name for name in mp_holistic.FaceLandmark.__members__]
|
420 |
hand_landmarks = [name for name in mp_holistic.HandLandmark.__members__]
|
421 |
|
422 |
def detect_landmarks(image_path):
|
@@ -450,7 +449,7 @@ def detect_landmarks(image_path):
|
|
450 |
for idx, landmark in enumerate(results.face_landmarks.landmark):
|
451 |
x = int(landmark.x * image_width)
|
452 |
y = int(landmark.y * image_height)
|
453 |
-
keypoints.append(('
|
454 |
|
455 |
# Left hand landmarks
|
456 |
if results.left_hand_landmarks:
|
@@ -486,6 +485,14 @@ def apply_touch_points(image_path, keypoints):
|
|
486 |
|
487 |
return image_pil
|
488 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
489 |
|
490 |
# Function to create a sensation map
|
491 |
def create_sensation_map(width, height, keypoints):
|
|
|
416 |
|
417 |
# Lists of landmark names for pose, face, and hands
|
418 |
pose_landmarks = [name for name in mp_holistic.PoseLandmark.__members__]
|
|
|
419 |
hand_landmarks = [name for name in mp_holistic.HandLandmark.__members__]
|
420 |
|
421 |
def detect_landmarks(image_path):
|
|
|
449 |
for idx, landmark in enumerate(results.face_landmarks.landmark):
|
450 |
x = int(landmark.x * image_width)
|
451 |
y = int(landmark.y * image_height)
|
452 |
+
keypoints.append(('FACE_LANDMARK_' + str(idx), (x, y)))
|
453 |
|
454 |
# Left hand landmarks
|
455 |
if results.left_hand_landmarks:
|
|
|
485 |
|
486 |
return image_pil
|
487 |
|
488 |
+
# Example usage
|
489 |
+
image_path = 'path_to_your_image.jpg'
|
490 |
+
keypoints = detect_landmarks(image_path)
|
491 |
+
if keypoints:
|
492 |
+
output_image = apply_touch_points(image_path, keypoints)
|
493 |
+
output_image.show()
|
494 |
+
else:
|
495 |
+
print("No landmarks detected in the image.")
|
496 |
|
497 |
# Function to create a sensation map
|
498 |
def create_sensation_map(width, height, keypoints):
|