This project was done as a part of the data collection project of Topics in Deep Learning(TDL).
The Yoga-Pose Dataset is a comprehensive collection of yoga pose images, pose landmarks, and calculated angles for various yoga poses. This dataset serves as a valuable resource for researchers, practitioners, and developers interested in yoga pose detection, recognition, and analysis. Each yoga pose in the dataset is meticulously annotated with pose landmarks and corresponding angles, providing rich ground truth data for machine learning model training and evaluation.
The code utilizes the MediaPipe library for pose estimation, which provides a pre-trained pose detection model. It processes images containing individuals performing yoga poses and extracts the pose landmarks, which are then used to calculate the angles between different body joints.
The dataset consists of the following components:
-
Images: High-quality images of different yoga poses captured from various perspectives and angles.
-
Pose Landmarks: Pose landmarks extracted using the MediaPipe Pose model, providing key points representing different body parts in each yoga pose image.
-
3D Models: Three-dimensional (3D) models generated from the pose landmarks, allowing visualization of the yoga poses in a three-dimensional space.
-
Calculated Angles: Angles between specific body parts in each yoga pose, providing detailed insights into the pose dynamics and alignment.
The dataset covers a wide range of yoga poses, including:
- ArdhaChandrasana (Half-Moon)
- Downward Dog
- Triangle
- Veerabhadrasana
- Natarajasana
- Vrukshasana
- BaddhaKonasana
- UtkataKonasana
- Elbow Angles
Left Elbow Angle: Angle between the left shoulder, elbow, and wrist points.
Right Elbow Angle: Angle between the right shoulder, elbow, and wrist points.
- Shoulder Angles
Left Shoulder Angle: Angle between the left elbow, shoulder, and hip points.
Right Shoulder Angle: Angle between the right hip, shoulder, and elbow points.
- Knee Angles
Left Knee Angle: Angle between the left hip, knee, and ankle points.
Right Knee Angle: Angle between the right hip, knee, and ankle points.
- Additional Angles
Angle for ArdhaChandrasana 1: Angle specific to ArdhaChandrasana, calculated between ankle, hip, and opposite ankle points.
Angle for ArdhaChandrasana 2: Second angle specific to ArdhaChandrasana, calculated between ankle, hip, and opposite ankle points.
Hand Angle: Angle between the left elbow, right shoulder, and right elbow points.
Left Hip Angle: Angle between the left shoulder, left hip, and left knee points.
Right Hip Angle: Angle between the right shoulder, right hip, and right knee points.
- Clone the repository:
git clone https://github.com/Manoj-2702/TDL-Dataset.git
- Navigate to the project directory:
cd TDL-Dataset
- Install the required dependencies:
pip install -r requirements.txt
- Add images to the required pose to the TRAIN Folder and the respective pose folder
-
Ensure that your yoga pose images are stored in the TRAIN/ directory within the project folder.
-
Run the script main.py to process the images and collect the pose data:
python main.py
This dataset is used for training deep learning models for detecting yoga poses from an image. Researchers, developers, and practitioners can utilize this dataset for various purposes, including:
- Training and evaluating machine learning models for yoga pose detection and recognition.
- Conducting experiments and studies on yoga pose dynamics and alignment.
- Developing applications and tools for yoga practitioners to improve their practice.
There will be two datasets which will be created. One containing the labels and the angles calculated. The other dataset contains the mediapipe landmarks for all the images.
Contributions to this project are welcome. If you encounter any issues or have suggestions for improvements, feel free to open an issue or create a pull request.