google/next-prediction

could i only use this to predict the action by only input the personfeature and keypoint ?

shaldor0315 opened this issue · 8 comments

could i only use this to predict the action by only input the personfeature and keypoint ?

You'll need to change the code in models.py by removing other features in enc_h_list and enc_last_state_list. Also, you need to remove the trajectory decoder. Then re-train the model.