行人姿态估计源码AlphaPose

机器环境:ubuntu16.04,cuda8.0

1.torch&tensorflow 版本

源码地址:https://github.com/MVIG-SJTU/AlphaPose

首先跑的是torch版本,安装了torch和tensorflow后运行程序出现如下错误:

Error in `python2': corrupted size vs. prev_size: 0x0000000002981e70

可见issue"https://github.com/MVIG-SJTU/AlphaPose/issues/88"

放弃该版本,根据作者建议转向pytorch版本。

2.pytorch版本

首先安装pytorch,源码使用的是python3,我的机器系统是ubuntu16.04,自带python3.5,因此首先安装python3.5对应的pytorch。pytorch官网:https://pytorch.org/。因为网络问题,需要选择之前版本下载,如下图所示:

行人姿态估计源码AlphaPose_第1张图片

选择python3.5 和 cuda 8.0对应的版本:

行人姿态估计源码AlphaPose_第2张图片

安装之后,按照源码后续步骤下载预训练模型即可。

 

你可能感兴趣的:(行人姿态估计源码AlphaPose)