Ubuntu18.04下安装配置alphapose(pytorch 版本)

系统为Ubuntu18.04,cuda版本为10.1。

登录https://github.com/MVIG-SJTU/AlphaPose/,选择pytorch分支,clone下载

git clone https://github.com/MVIG-SJTU/AlphaPose.git

安装python3.6必要的依赖项,默认在requirements.txt,可以根据需要安装,其中最后一个ntpath无需安装。安装的torch时可以登录官网,根据系统情况旋转版本,本次为最新的版本

pip3 install torch torchvision

手动下载duc_se.pth和 yolov3-spp.weights两个model分别到./models/sppe 及 ./models/yolo 下。

为避免out of memory问题,修改所使用的模型cfg文件中的subdivision的参数,将subdivisions=8改成subdivisions=64。

 

Leave a Reply

您的电子邮箱地址不会被公开。 必填项已用 * 标注

You may use these HTML tags and attributes:

<a href="" title=""> <abbr title=""> <acronym title=""> <b> <blockquote cite=""> <cite> <code> <del datetime=""> <em> <i> <q cite=""> <s> <strike> <strong>