复现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下后重新测试运行。