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

self_attention #7

Closed imuxwb closed 6 months ago

imuxwb commented 9 months ago

I want to know where the self-attention was added, and the self-attention matrix graph, is it calculated using the bias matrix

fokhruli commented 6 months ago

Self attention was calculated using convlstm network which calculate spatial and temopral attention of each joint. You can found the code description in sgcn.py file.