nkliuyifang / 2s-agcn-skeleton-based-action-recognition

2s-AGCN-Skeleton-Based-Action-Recognition

版本

pytorch 1.8.1
注：需要安装tensorboardX库： pip install tensorboardX

数据准备

• npy数据转json文件
  python numpy2json.py
• 数据处理 (注意要使用cd改变当前路径)
  cd data_gen
python kinetics_gendata.py
cd ..
  
• 生成bone的数据 (注意要使用cd改变当前路径)
  cd data_gen
python gen_bone_data.py
cd ..
  

训练和测试：

• 训练
  python main.py --config ./config/kinetics-skeleton/test_joint.yaml
python main.py --config ./config/kinetics-skeleton/test_bone.yaml
  

• 测试
  python main.py --config ./config/kinetics-skeleton/train_joint.yaml
python main.py --config ./config/kinetics-skeleton/train_bone.yaml
  

• 整合
  python ensemble.py

• 整个测试流程:
  先分别test：
  python main.py --config ./config/kinetics-skeleton/test_joint.yaml
python main.py --config ./config/kinetics-skeleton/test_bone.yaml
  再整合：
  python ensemble.py
  

参考文献：

[1] Shi L , Zhang Y , Cheng J , et al. Two-Stream Adaptive Graph Convolutional Networks for Skeleton-Based Action Recognition[J]. 2018.
[2] Zhang P , Lan C , Zeng W , et al. Semantics-Guided Neural Networks for Efficient Skeleton-Based Human Action Recognition[C]// 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, 2020.