Keypoint extraction code

[snalyami]

Thank you for your amazing work!
I would like to train the model on a custom dataset. Do you mind sharing the code to extract the keypoints for the GCN model.