参考了许多博文,慢慢地也就把st-gcn跑出来了,参考的文章一会附在文章里面,实测有用。
1.安装st-gcn
复现STGCN CPU版 (ubuntu16.04+pytorch0.4.0+openpose+caffe)_Significance的博客-程序员秘密复现旧版STGCN GPU版 (win10+openpose1.5.0)_Significance的博客-程序员秘密2
2.准备训练数据集
官方使用的是Kinetics-skeleton和NTU RGB+D两个数据集来训练网络,如果是自制数据集,我们需要将自己的训练集先转换为这两个数据集的格式,然后再转化为st-gcn所需要的格式。
那么如何将自己的训练集转换成这两个数据集的格式呢?
1)直接在st-gcn的程序里跑自己的数据
运行st-gcn:
python main.py demo_old --video E:/st-gcn/resource/media/数据集名称/running.mp4 --openpose E:/openpose/build/bin
跑出来的json文件可以在st-gcn文件里获取到,单帧和一个视频所有帧都有。文件位置在data/openpose_estimation,data里是所有帧json文件,snippets是单帧json文件。
2)更改标签与标签序号
此时,存在的问题是json文件末尾的"label":"unknow"&