Exercise Counter

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).

Demo Videos

Steps to run

  1. Clone this repository
git clone https://github.com/Garvit-32/exercise-counter.git  
  1. Install all the dependencies
pip install -r requirements.txt   
  1. Download weights from here and put in folder named weights

  2. 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 

DEMO

References

  1. https://github.com/stefanopini/simple-HRNet
  2. https://github.com/duc-tran/infant