本文为 AAAI 2018 录用论文「Spatial Temporal Graph Convolutional Networks for Skeleton Based Action Recognition」,香港中文大学提出一种时空图卷积网络,并利用它们进行人类行为识别。这种算法基于人类关节位置的时间序列表示而对动态骨骼建模,并将图卷积扩展为时空图卷积网络而捕捉这种时空的变化关系。
文章链接:https://arxiv.org/abs/1801.07455
Github 代码:https://github.com/yysijie/st-gcn?
由于该项目是基于openpose实现的,环境的配置比较复杂,为了顺利跑起程序,在docker_hub试着搜索有没有相关镜像,果然有
直接拉去镜像
docker pull heegreis/st-gcn:2080ti
创建相关容器
nvidia-docker run --name=st-gcn -i -t heegreis/st-gcn:2020ti /bin/bash
进入到workspace目录
下载st-gcn项目
git clone https://github.com/yysijie/st-gcn
进入到models目录,进入作者提供的google网盘下载需要的模型
返回到workspace目录
CUDA_VISIBLE_DEVICES='0,1' python3 main.py demo_old --openpose '/openpose/build' --video '/workspace/resource/media/skateboarding.mp4'
报错 FFmpeg的问题,可能原因是ffmpeg版本过低的原因,由于pip list
also indicates the presence of ffmpeg-1.4
解决方案
sudo add-apt-repository ppa:mc3man/trusty-media sudo apt-get update sudo apt-get install ffmpeg sudo apt-get install frei0r-plugins
继续运行,一切正常,如下
展示效果