Yoga Pose Classifier
ONNX-exported CNN models trained on a 107-class yoga pose dataset.
Models
| File | Architecture | Val Accuracy |
|---|---|---|
yoga_baseline.onnx |
1× Conv + Flatten | XX.X% |
yoga_advanced.onnx |
3× Conv + BN + Pool | XX.X% |
yoga_vgg16.onnx |
VGG16 transfer learning | XX.X% |
Input: (1, 160, 160, 3), float32, pixel range [0, 255].
Output: (1, 107) softmax probabilities.
Usage
```python import onnxruntime as ort import numpy as np
session = ort.InferenceSession('yoga_advanced.onnx') output = session.run(None, {'input': image_array})[0] ```
Live demo
Training
See GitHub repo.