实战OpenPose项目3:pytorch 实现openpose(包括手和身体姿态估计)

简介

pytorch实施openpose包括身体和手姿态估计,并且pytorch模型直接从转换openpose caffemodel通过caffemodel2pytorch。如果你有兴趣,你可以用同样的方式实现人脸关键点检测。注意人脸关键点检测器是使用 [Simon et al. 2017] 手。

openpose 通过人体姿态估计的结果来检测手部,请参考https://github.com/CMU-Perceptual-Computing-Lab/openpose/blob/master/src/openpose/hand/handDetector.cpp的代码。在论文中,它指出:

This is an important detail: to use the keypoint detector in any practical situation, 
we need a way to generate this bounding box. 
We directly use the body pose estimation models from [29] and [4], 
and use the wrist and elbow position to approximate the hand location, 
assuming the hand extends 0.15 times the length of the forearm in the same direction.

具体请参考上一篇的上一篇文章:实战OpenPose项目

官网:https://github.com/Hzzone/pytorch-openpose

快速入门

安装要求
创建一个python 3.7环境,例如:

conda create -n pytorch-openpose python=3.7
conda activate pytorch-openpose
按照此处的快速入门指南安装 pytorch(使用 pip)https://download.pytorch.org/whl/torch_stable.html

用pip安装其他需求

pip install -r requirements.txt

下载训练好的模型

在这里插入图片描述三种下载链接如下:

*.pth 文件是pytorch模型,如果你想使用caffe作为后端,你也可以下载caffemodel文件。

下载pytorch模型,放到model项目根目录下的目录下

运行演示

开始运行demo推理:

python demo_camera.py
使用来自网络摄像头的提要运行演示或运行

python demo.py
使用图像文件夹中的图像或运行

python demo_video.py
处理视频文件(需要ffmpeg-python)。

骨骼
在这里插入图片描述
身体姿势估计
在这里插入图片描述
手姿势估计
在这里插入图片描述

参考:
https://github.com/Hzzone/pytorch-openpose

  • 11
    点赞
  • 110
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
OpenPose是一种基于深度学习的姿势估计算法,它可以通过分析图像或视频中人体的关键点位置来实现人体姿势的识别与重建。而PyTorch是一个广泛应用于深度学习的开源机器学习框架,它提供了丰富的工具和库,可简化模型的构建、训练和部署过程。 在使用PyTorch实现OpenPose时,我们可以使用PyTorch的张量操作、自动求导和并行计算功能来构建和训练网络模型。首先,我们需要将OpenPose的网络结构转化为PyTorch模型的形式,可以利用PyTorch提供的各类层和函数(如卷积层、池化层、Batch Normalization等)来搭建网络的基本结构。 接着,我们需要定义损失函数来衡量网络输出与真实姿势的差异,一般选择欧式距离或关节坐标误差等指标作为损失函数。然后,通过调整网络的权重和参数,使用梯度下降等优化算法来最小化损失函数,从而不断优化网络的预测能力。 在训练模型时,可以使用PyTorch提供的数据加载和预处理工具(如DataLoader和transforms)来加载和处理训练数据,同时可以使用PyTorch的GPU加速功能来加快训练速度。 训练完成后,我们可以将训练好的模型用于姿势估计任务中,通过将图像或视频输入网络模型,获取网络输出的关键点位置,并进行后续的姿势重建和应用。 总结来说,通过使用PyTorch实现OpenPose,我们能够利用PyTorch强大的深度学习能力和丰富的工具来简化OpenPose算法的实现和训练过程,从而实现高效准确的人体姿势估计。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

源代码杀手

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值