pipixin321/Two-Branch-Network-For-SMPL-based-Action-Recognition

SMPL Action Recognition Implementation

Python

SMPL Action Recognition Implementation

准备工作

本项目开源地址为：https://github.com/pipixin321/Two-Branch-Network-For-SMPL-based-Action-Recognition.git

推荐的环境

Python 3.7
Pytorch 1.7
CUDA 10.1
依赖包

pip install -r ./requirements.txt

准备数据

方法一：
使用已经提取好的姿态参数：https://pan.baidu.com/s/1CPZ8CN1nY9RgVCn4OllpSQ?pwd=lbaf

方法二:
1.拍摄视频，视频命名格式为axx_cxx.mp4

如动作0，样本1：a00_c01.mp4

2.使用ROMP提取姿态参数

ROMP: https://github.com/Arthur151/ROMP，
input: video_folder
output: ideo_romp_result

数据预处理

python data_preprocess.py

input:./video_romp_result
output:./window_data

训练和测试模型

训练gcn部分

python main.py --mode train  --train_model gcn

训练cnn部分

python main.py --mode train  --train_model cnn

测试

python main.py --mode infer

APP的使用

python app.py

单击本地的URL链接进入ui界面、
将examples文件夹中的姿态参数序列上传并提交