前提：ubuntu下将python3.5.2设为默认（百度）

一.下载stgcn

（gitbub上fork后导入到gitee快些）: st-gcn: Spatial Temporal Graph Convolutional Networks (ST-GCN) for Skeleton-Based Action Recognition in PyTorch博客：https://blog.csdn.net/weixin_42661709/article/details/105056325

二.开始配置

1.安装pytorch0.4.0:

sudo pip3 install torch==0.4.0 -f https://download.pytorch.org/whl/cup/stable -i Simple Index

2.openpose pythonAPI

ubuntu16.04 openpose cpu版 python3_I'm you.的博客-CSDN博客

3.下载依赖包：

sudo apt-get install ffmpeg

pip3 install -r requirements.txt

4.安装

cd st-gcn
cd torchlight;
sudo python3 setup.py install;
cd ..

5.下载模型

法一(相当慢)：bash tools/get_models.sh

法二：从别人的网盘下载(写文章-CSDN博客)

pose_iter_440000.caffemodel放到st-gcn/models/pose/coco中

st_gcn.kinetics.pt等文件放到 st-gcn/models中

6.运行Demo（配置差可以试个短视频就几帧那种。想跑图片就把.jpg改成.mp4）

python main.py demo [--video ${PATH_TO_VIDEO}] [--openpose ${PATH_TO_OPENPOSE}]

新版STGCN：python3 main.py demo --openpose '/home/chq/openpose/build' --video '/home/chq/st-gcn/resource/media/skateboarding.mp4'

旧版：python3 main.py demo_old --video /home/chq/st-gcn/resource/media/test.mp4 --openpose /home/chq/openpose/build

旧版默认保存在 ./data/demo_result/下，也可以自己设置保存位置如：--output_dir /home/chq/st-gcn/resource/media_out

第一个图是10秒的视频我跑了两天.....第二个图是改图片后缀了测试的图片

6.解决错误：

1.cpu版的cuda问题

device = torch.device("cuda" if args.cuda else "cpu")。将io.py中103行True修改为False

2.RuntimeError: Error(s) in loading state_dict for Model

问题解决：
原因是作者提供(bash tools/get_models.sh)的模型参数—— st_gcn.kinetics.pt 有问题。所以要下载从另外的途径下载可用的文件https://pan.baidu.com/s/1JrfpRqt0uF8AydiJuxANZw#list/path=%2F&parentPath=%2Fsharelink53688367-280314025530152

(password: j4rt )。放到models下后重新测试运行。

任何程序错误，以及技术疑问或需要解答的，请扫码添加作者VX : 18565453898

复现STGCN CPU版（ubuntu16.04+pytorch0.4.0+openpose+caffe）相关推荐

Ubuntu16.04+GT720M + Cuda8.0+py-faster-rcnn(caffe)
Ubuntu16.04 + GT 720M + Cuda8.0+ Py-faster-rcnn(caffe) Cuda8.0 安装 cuda 的安装其实挺简单的,有两种安装方法,一种使用deb文件,另 ...
ubuntu16.04 在cuda9.0环境下编译安装opencv2.4.13.7
ubuntu16.04 在cuda9.0环境下编译安装opencv2.4.13.7 安装步骤: 1.安装cuda9.0:https://blog.csdn.net/zhuangwu116/articl ...
ubuntu16.04 安装cuda9.0+cudnn7.0.5+tensorflow+nvidia-docker配置GPU服务
[摘要] docker很好用,但是在GPU服务器上使用docker却比较复杂,需要一些技巧,下面将介绍一下在ubuntu16.04环境下的GPU-docker环境搭建过程. 第一步: 删除之前的nvi ...
Ubuntu16.04 安装 OpenMPI4.0.0
Ubuntu16.04 安装 OpenMPI4.0.0 文章目录 OpenMPI 简介一.OpenMPI 1.0.0版本下载二.OpenMPI 安装步骤 1.解压并进行配置 2.Build 并安装 ...
ubuntu16.04+gtx1080ti+cuda10.0+cudnn7.4.2/cudnn7.6.4 环境配置
配置:ubuntu16.04+gtx1080ti+cuda10.0+cudnn7.4.2/cudnn7.6.4+显卡驱动NVIDIA-Linux-x86_64-410.78 tensorflow1.1 ...
caffe+CPU︱虚拟机+Ubuntu16.04+CPU+caffe安装笔记
由于本机是window10系统,所以想尝试caffe就在自己电脑上整了一个虚拟机(详情可见:win10系统搭建虚拟机:VMware Workstation Player 12环境+Ubuntu Kyl ...
Ubuntu16.04编译android6.0.1源码记录
目录目录一.安装环境二.下载源码 1.下载repo 2.初始化repo 3.同步源代码关于驱动三.编译源码四.导入源码到AS 五.刷入真机六.修改源码总结: 3.同步源代码关于驱动 ...
编译内核启用KASan动态检测内核内存错误功能(ubuntu16.04 4.4.0内核编译升级到linux-4.4.252版本)
目录一.编译内核并配置支持KAsan 1.下载源码 2.安装编译内核必须的库 3.清理之前内核编译生成及相关配置 4.配置内核参数 5.开始编译 6.模块(驱动)安装 7.安装新编译的内核 8.修改 ...
ubuntu16.04+GTX1060+ CUDA9.0+CUDANN7.0 tensorflow(GPU)安装，实测有效，安装包我放在云盘中，直接下载使用。
最近在运行一些 semantic segmentation 的代码,好多都是GPU版本的,自己就安装一下,记录一下,开始错了好多次,最后参考了一些文章,发现这个版本问题很重要,我用的是一下版本cuda ...

复现STGCN CPU版（ubuntu16.04+pytorch0.4.0+openpose+caffe）

二.开始配置

6.解决错误：

复现STGCN CPU版（ubuntu16.04+pytorch0.4.0+openpose+caffe）相关推荐

最新文章

热门文章

复现STGCN CPU版 （ubuntu16.04+pytorch0.4.0+openpose+caffe）

二.开始配置

6.解决错误：

复现STGCN CPU版 （ubuntu16.04+pytorch0.4.0+openpose+caffe）相关推荐

最新文章

热门文章

复现STGCN CPU版（ubuntu16.04+pytorch0.4.0+openpose+caffe）

复现STGCN CPU版（ubuntu16.04+pytorch0.4.0+openpose+caffe）相关推荐