---
license: apache-2.0
---


Pose Estimation Model

Overview

This project implements a Pose Estimation Model using MediaPipe, OpenCV, and NumPy to analyze human body movement from images and videos. It extracts key body landmarks, calculates angles between joints, and logs the data for further analysis.

Features

Pose Detection: Detects and extracts body landmarks from images and videos.

Angle Calculation: Computes angles between body joints to analyze posture.

Distance Measurement: Measures distances between key body points.

Video Processing: Processes videos frame by frame and logs body data.

Data Logging: Saves pose estimation data to CSV for analysis.

Model Configuration: Supports saving and loading model configurations in JSON.

Dependencies

Ensure you have the following Python libraries installed:

pip install opencv-python mediapipe pandas matplotlib numpy

Installation

Clone the repository and navigate to the project directory:

git clone : 
cd pose-estimation

Usage

1. Initialize the Model

from pose_estimation import PoseEstimationModel
model = PoseEstimationModel()

2. Process an Image

import cv2
image = cv2.imread('sample_image.jpg')
output_image, landmarks = model.detect_pose(image, model.pose_video, display=True)

3. Process a Video

video_path = 'input_video.mp4'
log_file = 'output_log.csv'
model.process_video(video_path, log_file)

4. Calculate Body Angles

angles = model.body_angles(landmarks)
print(angles)

5. Save & Load Model Configuration

model.save_model_config('model_config.json')
loaded_model = PoseEstimationModel.load_model_config('model_config.json')

Output

The processed image/video with pose landmarks overlaid.

A CSV file containing body joint angles and measurements.

Applications

Sports Analytics: Analyze athlete posture and movement.

Health & Fitness: Assist in exercise form correction.

Rehabilitation: Monitor patient recovery through movement tracking.

Gaming & AR: Use body pose as input for interactive applications.

License

This project is licensed under the apache-2.0 License.