Count the repetitions of exercise using pose estimation implemented in Pytorch and trained on HRNet (Link) .Model is also scalable to multiple person using YOLOv3. This repo is built on the top of this repo (Link).
- Clone this repository
git clone https://github.com/Garvit-32/exercise-counter.git
- Install all the dependencies
pip install -r requirements.txt
-
Download weights from here and put in folder named weights
-
Run inference
python main.py --filename <path to exercise video> --exercise_type <exercise number> --weights <path to weights>
| Exercise Name | Exercise Number |
|---|---|
| PushUps | 1 |
| Squats or Situps | 2 |
| Pullups | 3 |
| Dumbell Curl | 4 |
| Dumbell Side Lateral | 5 |
| It can be extended to other exercise as well. |
Example
python main.py --filename 'demo_videos/squat.mp4' --exercise_type 2
 |
 |
 |
 |