vaibhavmalhotra-ai/yoga-pose-recognition

🧘 Yoga Pose Recognition

Powered by CNN, Transfer Learning & Explainable AI.

Upload a yoga pose image and the model identifies the posture, then optionally reveals how it made that decision using Grad-CAM.

What it does

• —Pose prediction — a fine-tuned ResNet50 classifies the uploaded image into one of the trained yoga pose classes, with a confidence score.
• —Explainability (Grad-CAM) — click "Yes, Show Me!" to see which regions of the image the model focused on, via a Grad-CAM heatmap and a blended focus overlay.

Color guide for the heatmap

• —🔴 Red → strongest attention
• —🟡 Yellow → important region
• —🔵 Blue → less important region

Tech stack

• —Model: ResNet50 (transfer learning) saved as resnet50_best.keras
• —Explainability: Grad-CAM (Gradient-weighted Class Activation Mapping)
• —Interface: Gradio
• —Image processing: OpenCV, Pillow, NumPy

Author

Vaibhav Malhotra