OpenPose训练自己的数据集模型Realtime Multi-Person Pose Estimation

本文档详细介绍了如何使用OpenPose训练自己的数据集进行实时多人姿势估计,包括下载官方代码、获取数据集、安装MATLAB、编译Caffe、数据预处理、准备模型文件和模型训练等步骤,特别提到了在编译和训练过程中可能遇到的错误及其解决方案。

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

(持续更新中)

目录

1.下载训练openpose模型的相关官方代码

2.下载训练所需要的数据集

3.安装MATLAB

4.编译训练openpose模型所用的caffe

5.训练数据集的准备及处理

6.准备训练用的prototxt模型文件

7.模型训练


1.下载训练openpose模型的相关官方代码

GitHub地址:

https://github.com/ZheC/Realtime_Multi-Person_Pose_Estimation

git或者下载解压之后得到如下内容:

 

2.下载训练所需要的数据集

cd training 

./getData.sh

运行上述脚本,下载训练需要的数据集,同时还git下了一个cocoAPI。

下载时间较长,因为数据集共有40G左右。

 

3.安装MATLAB

由于步骤1中得到的数据集处理代码中需要用到matlab,因此需要安装一下。

Ubuntu上的matlab安装参考这篇文章:

https://blog.youkuaiyun.com/qq_38469553/article/details/82050851

 

4.编译训练openpose模型所用的caffe

GitHub地址:

https://github.com/CMU-Perceptual-Computing-Lab/caffe_train

下载之后得到caffe_train文件夹,这个文件夹中和原始的caffe相比在源码中添加了一些openpose相关的内容。

将此caffe文件夹中的Makefile.config.example文件copy一份为Makefile.config文件,之后按照需要修改其中的内容(和编译原始的caffe基本相同)。

具体请参考:

https://blog.youkuaiyun.com/qq_38469553/article/details/78786892

这篇文章中关于caffe配置的相关内容。PS:如果你的ubuntu系统之前没有成功编译通过caf

评论 39
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值