欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Windows10安装AlphaPose(不需要VS版)

程序员文章站 2024-03-22 13:35:10
...

最新版地址:AlphaPose最新版地址
code地址:AlphaPose安装

安装

  1. 下载安装代码
git clone -b pytorch https://github.com/MVIG-SJTU/AlphaPose.git
  1. 安装依赖包
pip install -r requirements.txt

在这一步,安装torch==0.4.0遇到ERROR: No matching distribution found for torch==0.4.0报错。
首先查看了一下当前电脑里的torch版本。

torch.__version__
# 0.4.1

当前版本是0.4.1,于是首先卸载这个版本的torch

pip uninstall torch

然后根据这篇博客的讲解,从【http://download.pytorch.org/whl/cu90/torch-0.4.0-cp36-cp36m-win_amd64.whl】下载【torch-0.4.0-cp36-cp36m-win_amd64.whl】,然后cmd切换到.whl文件目录下,执行

pip install torch-0.4.0-cp36-cp36m-win_amd64.whl

即可安装成功。

这里记录一下各种安装包.whl下载地址:Python Extension Packages

  1. 下载已经训练好的模型duc_se.pth(BaiDu Yun)放到./models/sppe文件夹下,再下载yolov3-spp.weights(BaiDu Yun)放到./models/yolo文件夹下。

测试

  1. 对图片进行测试
python3 demo.py --indir ${img_directory} --outdir examples/res 

因为我的电脑只有一个python,且只能执行单线程任务,所以我的测试代码是

python demo.py --indir examples/demo --outdir examples/res --sp

Windows10安装AlphaPose(不需要VS版)
输出结果是一个记录关键点坐标的json文件。

  1. 对视频进行测试
python3 video_demo.py --video ${path to video} --outdir examples/res --save_video

同样的原因,修改为

python video_demo.py --video examples/demo/video.avi --outdir examples/res --save_video --sp

Windows10安装AlphaPose(不需要VS版)
完整视频:BaiDu Yun
提取码:rwfd

  1. Webcam测试
python webcam_demo.py --webcam 0 --outdir examples/res --vis
相关标签: 文章代码实现