本文為 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
繼續運行,一切正常,如下
展示效果