系统为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。