最新版地址:AlphaPose最新版地址
code地址:AlphaPose安装
安装
- 下载安装代码
git clone -b pytorch https://github.com/MVIG-SJTU/AlphaPose.git
- 安装依赖包
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
- 下载已经训练好的模型
duc_se.pth(BaiDu Yun)放到./models/sppe文件夹下,再下载yolov3-spp.weights(BaiDu Yun)放到./models/yolo文件夹下。
测试
- 对图片进行测试
python3 demo.py --indir ${img_directory} --outdir examples/res
因为我的电脑只有一个python,且只能执行单线程任务,所以我的测试代码是
python demo.py --indir examples/demo --outdir examples/res --sp

输出结果是一个记录关键点坐标的json文件。
- 对视频进行测试
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

完整视频:BaiDu Yun
提取码:rwfd
- Webcam测试
python webcam_demo.py --webcam 0 --outdir examples/res --vis
本文详细介绍AlphaPose的最新版地址、代码安装流程,包括解决torch版本冲突问题,以及如何下载预训练模型和测试代码。提供了对图片和视频进行人体姿态估计的步骤,包括输出关键点坐标JSON文件的方法。
383

被折叠的 条评论
为什么被折叠?



