fokhruli / STGCN-rehab

This repository provides training and evaluation code for paper titled "Graph Convolutional Networks for Assessment of Physical Rehabilitation Exercises" (accepted in IEEE TNSRE)
https://fokhruli.github.io/STGCN-rehab/
MIT License
24 stars 6 forks source link

General questions #1

Closed BenTheQ closed 2 years ago

BenTheQ commented 2 years ago
fokhruli commented 2 years ago

Please, find reply of queries below:

1) The data processing step was done following the KIMORE dataset paper and github. Then we use ''data_processing.py'' file for pre-processing the joint data. For each joint, we have (x,y,z) position along the z axis, joint number along y axis and finally timestep along x axis. The ''data_processing.py'' file will output the aforementioned data structure.

2) We added preprocessed dataset link in github repo. Please, check it now.

3) Yes, for X_train we only use the joint position. However, you can also use joint orientation along with the joint position to make the model robust.

4) Since the exercises differ from each other in various form, we train the network for each of the exercises separately.

5) ''best_model.hdf5'' is the weight of our network for the best performing model among 1500 epochs. Yes, it was for KIMORE ex5. But you can find all exercise here.

We will update and organize (with rgb evaluation code) our github repository very soon.