OpenPose -- stage1

本文介绍在Ubuntu 16.04系统下搭建TensorFlow人体姿态估计环境。先安装TensorFlow GPU版和CUDA,激活环境后安装OpenCV,从GitHub拉取源码,安装Swig并进行代码处理。最后给出测试命令,可对图片、视频进行测试,还能调用摄像头实时获取视频。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

环境:Ubuntu16.04

           Anaconda3

           tensorflow 1.9.0

           CUDA 9.0

首先安装tensorflow gpu版本,具体过程请参考链接【Tensorflow】(一):环境配置(Anaconda)。

CUDA的安装过程请参考链接:【CUDA】linux ubuntu16.04 卸载cuda8.0 安装cuda9.0。

安装完成之后激活tensorflow环境,使用下面的命令:

source activate tensorflow_gpu
接下来还需要安装opencv,具体的安装过程请参考链接:【Tensorflow】tensorflow中添加opencv支持。

下面就可以从github上拉取源码了。

git clone git@github.com:ildoonet/tf-pose-estimation.git
cd tf-pose-estimation
pip install -r requirements.txt
下面还需要swig支持,所以

cd tf_pose/pafprocess
安装swig,

sudo apt install swig
现在可以swig代码了,

swig -python -c++ pafprocess.i && python3 setup.py build_ext --inplace
环境已经搭建好了,下面可以进行测试,

python run.py --model=mobilenet_thin --resize=432x368 --image=./images/p2.jpg


 另外还可以测试视频,

​python run_video.py --model=mobilenet_thin --resolution=432x368 --video=test.mp4
或者调用摄像头,实时获取视频,

python run_webcam.py --model=mobilenet_thin --resolution=432x368 --camera=0
 

### OpenPose 网络架构及原理 #### 一、概述 OpenPose 是一种广泛应用于人体姿态估计的技术,能够实时检测多人的关键点位置并构建骨架连接。该技术不仅限于单人场景,在复杂背景下也能保持较高的准确性[^1]。 #### 二、网络架构特点 OpenPose 的核心在于采用了自底向上的方法(bottom-up),即先独立识别图像中的各个身体部位(如手肘、膝盖等),再通过关联字段(Part Affinity Fields, PAFs)将这些局部特征组合成完整的个体姿态模型。 - **卷积神经网络(CNN)**:作为基础框架,用于提取输入图片的高层次语义信息; - **Stage 模块**:由多个相同结构的子网组成,每个stage负责预测一次关键点热力图和PAFs;随着层数加深,精度逐渐提高。 - **跳跃链接(Skip Connections)**:为了保留更多细节,不同层次间存在短路连接机制,使得浅层特征可以直接传递给深层部分参与计算。 ```python import cv2 from openpose import pyopenpose as op params = dict() params["model_folder"] = "./models/" opWrapper = op.WrapperPython() opWrapper.configure(params) datum = op.Datum() imageToProcess = cv2.imread("test.jpg") datum.cvInputData = imageToProcess opWrapper.emplaceAndPop([datum]) print("Body keypoints: \n" + str(datum.poseKeypoints)) ``` #### 三、工作流程解析 整个过程大致分为两个主要步骤: - 预测阶段:利用训练好的CNN模型对输入图像进行前向传播运算,得到每种可能的身体部件对应的置信度分布(heatmaps)以及它们之间的相对位移关系(PAFs)[^2]; - 解析阶段:基于上述两组数据,采用贪心算法寻找最佳匹配方案来重建每个人物实例的位置与形态。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值