Spaces:
Sleeping
Sleeping
| import cv2 | |
| import mediapipe as mp | |
| import numpy as np | |
| import gradio as gr | |
| # Initialize MediaPipe Pose and Drawing utilities | |
| mp_pose = mp.solutions.pose | |
| pose = mp_pose.Pose(min_detection_confidence=0.5, min_tracking_confidence=0.5) | |
| mp_drawing = mp.solutions.drawing_utils | |
| # Utility function to calculate angles | |
| def calculate_angle(a, b, c): | |
| a, b, c = np.array(a), np.array(b), np.array(c) | |
| radians = np.arctan2(c[1] - b[1], c[0] - b[0]) - np.arctan2(a[1] - b[1], a[0] - b[0]) | |
| angle = np.abs(radians * 180.0 / np.pi) | |
| return angle | |
| # Function to provide plank feedback based on landmarks | |
| def check_plank_feedback(landmarks): | |
| shoulder = [landmarks[mp_pose.PoseLandmark.LEFT_SHOULDER.value].x, | |
| landmarks[mp_pose.PoseLandmark.LEFT_SHOULDER.value].y] | |
| hip = [landmarks[mp_pose.PoseLandmark.LEFT_HIP.value].x, | |
| landmarks[mp_pose.PoseLandmark.LEFT_HIP.value].y] | |
| ankle = [landmarks[mp_pose.PoseLandmark.LEFT_ANKLE.value].x, | |
| landmarks[mp_pose.PoseLandmark.LEFT_ANKLE.value].y] | |
| angle = calculate_angle(shoulder, hip, ankle) | |
| accuracy = max(0, min(100, (1 - abs(angle - 175) / 15) * 100)) | |
| feedback = "Good Plank Position" if 165 <= angle <= 180 else "Incorrect Plank" | |
| if angle < 160: | |
| feedback += " - Hips Too Low (Sagging)" | |
| elif angle > 180: | |
| feedback += " - Hips Too High" | |
| return feedback, int(accuracy) | |
| # Function to draw an accuracy bar on the frame | |
| def draw_accuracy_bar(image, accuracy): | |
| bar_x, bar_y = 50, 400 | |
| bar_width, bar_height = 200, 20 | |
| fill_width = int((accuracy / 100) * bar_width) | |
| cv2.rectangle(image, (bar_x, bar_y), (bar_x + bar_width, bar_y + bar_height), (200, 200, 200), 2) | |
| cv2.rectangle(image, (bar_x, bar_y), (bar_x + fill_width, bar_y + bar_height), (0, 255, 0), -1) | |
| cv2.putText(image, f"Accuracy: {accuracy}%", (bar_x, bar_y - 10), | |
| cv2.FONT_HERSHEY_DUPLEX, 0.6, (255, 255, 255), 2) | |
| # Main function to analyze plank form in a video | |
| def analyze_plank(video_path): | |
| cap = cv2.VideoCapture(video_path) | |
| frame_width, frame_height = int(cap.get(3)), int(cap.get(4)) | |
| fps = cap.get(cv2.CAP_PROP_FPS) if cap.get(cv2.CAP_PROP_FPS) > 0 else 30 | |
| fourcc = cv2.VideoWriter_fourcc(*'mp4v') | |
| output_video = "output_plank.mp4" | |
| out = cv2.VideoWriter(output_video, fourcc, fps, (frame_width, frame_height)) | |
| while cap.isOpened(): | |
| ret, frame = cap.read() | |
| if not ret: | |
| break | |
| image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) | |
| results = pose.process(image) | |
| image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR) | |
| if results.pose_landmarks: | |
| mp_drawing.draw_landmarks(image, results.pose_landmarks, mp_pose.POSE_CONNECTIONS) | |
| landmarks = results.pose_landmarks.landmark | |
| feedback, accuracy = check_plank_feedback(landmarks) | |
| draw_accuracy_bar(image, accuracy) | |
| color = (0, 255, 0) if "Good" in feedback else (0, 0, 255) | |
| cv2.putText(image, feedback, (50, 50), cv2.FONT_HERSHEY_COMPLEX, 1, color, 3) | |
| out.write(image) | |
| cap.release() | |
| out.release() | |
| return output_video | |
| # Gradio Interface for Plank Analysis | |
| gr.Interface( | |
| fn=analyze_plank, | |
| inputs=gr.Video(), | |
| outputs=gr.Video(), | |
| title="Plank Form Analyzer", | |
| description="Upload a video of your plank, and get feedback on your form!" | |
| ).launch() | |