AlphaPose是一款精准的多人姿态评估工具,并声称是第一款实时的开源系统。
工具支持TensorFlow和PyTorch两种实现。AlphaPose利用一种区域性的多人动作估计框架将不精准的人类边界框该进程精确的动作估计。这里有三种元素:对称空间转换网络(SSTN)、参数化姿态非极大抑制(NMS)以及姿态导向的生成器(PGPG)。
依赖:
Torch
TensorFlow(verson >= 1.2)
安装:
1.下载
git clone https://github.com/MVIG-SJTU/AlphaPose.git
cd AlphaPose/human-detection/lib/
make clean
make
cd newnms/
make
cd ../../../
2.安装Torch/TensorFlow
chmod +x install.sh
./install.sh
3.运行fetch_models.sh,下载预训练模型。
chmod +x fetch_models.sh
./fetch_models.sh
快速开始;
运行demo,
./run.sh --indir examples/demo/ --outdir examples/results/ --vis
AlphaPose既可以在图片、视频或多图中进行姿态估计,也能在画面中对动作进行追踪。它的输出形式非常广泛,包括PNG、JPG和AVI等具有关键点的图片形式,也有JSON格式的输出,这一特点也使其成为众多应用受欢迎的工具。
閱讀更多 AI踐行者 的文章