动作识别-论文及源码_st-gcn

  本文为 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  

继续运行,一切正常,如下

 

展示效果

 

posted @ 2020-09-08 09:18  刘文华  阅读(1714)  评论(0编辑  收藏  举报