動作識別-論文及源碼_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  

繼續運行,一切正常,如下

 

展示效果

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM