fokhruli / Real-Time-Rehabilitation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Real-Time-Rehabilitation-Interface-with-kinectv2

Real time rehabilitation assessment with kinect v2. We implement the spatio temporal graph convolutional neural network for rehabilitation exercise assessment according to the following paper: S. Deb, M. F. Islam, S. Rahman and S. Rahman, "Graph Convolutional Networks for Assessment of Physical Rehabilitation Exercises," in IEEE Transactions on Neural Systems and Rehabilitation Engineering, vol. 30, pp. 410-419, 2022, doi: 10.1109/TNSRE.2022.3150392.

How to run this code

First you need the Microsoft Kinect v2 and install the associated Kinect SDK. After that follow these steps:

  1. Install the requirements.txt file the command pip install -r requirements.txt.
  2. Download the model parameters from the google drive and store them in the best model folder.
  3. Run the PyKinectBodyGame_v1.py file with the command python PyKinectBodyGame_v1.py in the terminal.

Prediction

Output for the healthy person. The correctness score is shown in real time at the top middle of the screen. Consequently, exercise name is shown at the top left.

normal_demo_1.mp4
normal_demo_2.mp4

Output for the patient. The correctness score is shown in real time at the top middle of the screen. Consequently, exercise name is shown at the top right.

demo_patient_2.mp4
demo_patient_1.mp4

About

License:Apache License 2.0


Languages

Language:Python 100.0%