search

kenziyuliu / DGNN-PyTorch

Unofficial PyTorch implementation of the CVPR'19 paper "Skeleton-Based Action Recognition with Directed Graph Neural Networks".
Other
235 stars 59 forks source link

add node mask and action selector #27

Open cyun9601 opened 3 years ago

cyun9601 commented 3 years ago
  1. Correct the error to use .reshape for model learning in the latest version of pytorch.
  2. Added the function to learn using certain nodes only.
  3. Added the function to learn using specific actions only.