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

How can I train on selected classes? #7

Closed IssaIssa1 closed 5 years ago

IssaIssa1 commented 5 years ago

How can I train the DGNN on selected number of classes using NTU dataset instead in training it on all the data?

kenziyuliu commented 5 years ago

You may need to regenerate the data from the raw skeletons and filter out the unwanted classes, and train on the filtered data.

See this line in ntu_gen_joint_data.py:

action_class = int(filename[filename.find('A') + 1:filename.find('A') + 4])