复现STGCN CPU版 (ubuntu16.04+pytorch0.4.0+openpose+caffe)

前提:ubuntu下将python3.5.2设为默认(百度)

一.下载stgcn

(gitbub上fork后导入到gitee快些):

二.开始配置

1.安装pytorch0.4.0:

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

2.openpose pythonAPI

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

法二:从别人的网盘下载()

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

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

经验分享 程序员 微信小程序 职场和发展