分享好友 最新动态首页 最新动态分类 切换频道
8.2K Star,开源的AI虚拟健身教练你值得拥有!
2024-12-27 01:25

在人工智能技术更迭及后疫情时代的背景下,随着居民生活模式发生改变,智能运动健身行业得到了极大的发展。刘畊宏带起的“本草纲目”健身热潮也正式开启了居家健身时代,随之而来的便是人们对个性化、智能化的健身动作识别、矫正、计数的强烈需求,如何随时随地进行便捷、标准的健身运动逐渐成为了人们热议的话题。

8.2K Star,开源的AI虚拟健身教练你值得拥有!

在健身场景下,动作类型与人体骨骼点的位置变化之间存在着密不可分的关系,因此采取人体关键点检测技术实现智能化健身动作识别与计数就再为合适不过了。

图1: 基于PP-TinyPose的健身动作识别

那么具体如何能利用人体关键点检测实现智能化的健身动作识别与计数呢?今天就手把手带大家用飞桨搭建一个AI虚拟健身教练吧!

不慌,先码住完整教程:

https://github.com/PaddlePaddle/PaddleDetection

★ 记得Star收藏 ★

防止走丢又实时关注更新

项目难点

非常规姿态多

在健身、舞蹈等场景中,人体姿态明显区别于常见的站姿,存在较多复杂的姿态。常见的公开数据集中往往缺少这类数据,因此在处理这类复杂姿态时,容易出现检测框位置或关键点位置不准确的问题,导致动作识别出现误检漏检,无法达到令人满意的效果。

模型体积与速度要求高

体育健身动作持续时间通常在秒级以内,因此对检测响应速度要求极高,同时随着边缘端硬件在健身行业的普及,对模型的体积要求也愈加严苛。

视频抖动严重

使用关键点算法处理视频数据时,由于预测针对单帧图像进行,在视频结果上往往会有抖动的现象。对于依靠精细化坐标的应用场景(如健身计数、基于关键点的虚拟渲染等),非常容易造成误检或可视化效果不流畅的问题。

关键点点数需求多变

例如在芭蕾舞场景下,垫脚、勾脚等动作仅依赖脚腕关键点无法进行区分,往往需要更多骨骼点信息才能实现精细化动作识别。

方案设计

模型选型

PP-TinyPose的出世彻底打破了精度与速度无法兼得的难题,成为轻量级关键点检测的最佳选择。PP-TinyPose采用Top-Down的方式,先应用3.3M、150FPS的超轻量检测网络PP-PicoDet检测出人体,再用基于Lite-HRNet的移动端优化模型确保关键点检测的高精度,同时扩大数据集,减小输入尺寸,预处理与后处理加入AID、UDP和DARK等策略,保证模型的高性能。实现速度在FP16下达到122FPS的情况下,精度也可达到COCO数据集51.8%AP,不仅比其他类似实现速度更快,精度更是提升了130%!

数据准备

数据收集包含人体检测和人体关键点检测两部分,均使用COCO train2017和AI Challenger trainset作为训练集。将两数据集标注含义进行统一,并转换为COCO数据格式后执行训练。

图3:数据准备

模型训练优化

更令人惊喜的是,在原版的基础上,针对体育健身场景难点,PP-TinyPose分别从场景数据多样性、检测模型迭代、关键点稳定性三个方向全面升级,发布了专为体育动作识别打造的增强版本,最终相比原版在健身场景数据上提升9.1%AP,实现速度85FPS精度77.1%AP。同时,新增7个复杂动作的精准识别,包括侧身、卧躺、跳跃、高抬腿等非常规动作。具体来看:

场景数据多样性

在COCO数据集和AI Challenger数据集的基础上,进一步结合采集的健身、舞蹈等较难场景数据对模型进行迭代,提升模型预测效果。

图4:PP-TinyPose增强版效果对比图

检测模型迭代

使用PP-PicoDet增强版模型,在原版基础上引入TAL及ETA Head,优化PAN等结构,COCO数据集精度提升2个点以上。基于丰富的健身场景数据,人体检测模型精度提升6.8% mAP

关键点稳定性

加入OneEuro滤波器和EMA两种关键点稳定方式,将当前关键点坐标结果和历史关键点坐标结果结合计算,使得输出的坐标更加稳定平滑。

图5:稳定性升级对比图

如果模型仍无法满足要求,开发者还可以基于飞桨目标检测开发套件PaddleDetection所提供的二次开发文档,进行模型训练及优化、关键点新增、删减与类别自定义,5分钟即可实现关键点检测个性化开发。

图6:PP-TinyPose增强版支持手指、脚趾等29个关键点

  • 完整链接:

https://github.com/PaddlePaddle/PaddleDetection/blob/develop/docs/advanced_tutorials/customization/keypoint_detection.md

后处理逻辑

得到人体关键点检测结果后,可以结合不同关键点在相同帧的相对关系,或者不同帧下同一关键点的位置变化来进行动作识别和计数。

以侧平举为例,当手臂抬起角度接近水平,则认为手臂处于抬起动作;当手臂放下角度接近垂直,则认为手臂处于放下动作;两动作交替完成后认为侧平举动作计数加一。

除此之外,PP-TinyPose增强版还配套了深蹲和站姿上举的动作识别计数实现。

  • 详细代码链接:

https://github.com/zhiboniu/pose_demo_android/blob/release/1.0/app/src/main/cpp/pose_action.cc

端侧部署

图8:端侧部署效果展示

通过以上方法即可获得高效的健身动作识别计数模型,依托于飞桨端侧推理引擎Paddle Lite,能够灵活快速的部署到各类端侧设备上。开发者可以直接扫描下方二维码体验健身APP:

如需更新模型并实现更丰富的健身动作识别,可参考以下流程:

  1. 训练模型导出为.nb后缀的Paddle Lite模型;

  2. 替换pose_demo_android中models下的nb模型文件;

  3. 通过Android Studio编译运行代码,重新生成APP。

  • 详细步骤参考链接:

https://github.com/zhiboniu/pose_demo_android

作为轻量化骨骼点检测的开源SOTA算法, PP-TinyPose以精度速度高、微小目标识别效果好、多人检测速度快、端侧部署易用性强、体育健身动作检测准著称,在端侧微弱算力上,使用飞桨轻量化推理引擎Paddle Lite,即可快速实现各类健身动作识别与计数!

还在等什么,快来搭建属于你的AI健身教练吧!

  • 教程文档及模型下载链接:

最新文章
视频重绘工具DomoAI 不用SD视频一键就能转动漫
▼用户可以通过提供简短的文本提示,例如描述一位老巫师、一个在水下游泳的女孩或释放魔法的少女,来启动的艺术生成功能。这显示了的灵活性,使用户能够根据自己的需求和创意方向快速实现他们的想法。DomoAI致力于通过其工具台加速用户的实
谷歌的搜索建议XHR请求
是一种通过XMLHttpRequest对象发送异步HTTP请求的技术。XHR代表XMLHttpRequest,是一种在后台与服务器进行数据交换的技术,可以在不重新加载整个页面的情况下更新部分网页内容。XHR请求的优势包括:XHR请求的应用场景包括:腾讯云相关产品
解锁TikTok尾部达人力量:品牌信任在海外市场的构建之路
在数字营销浪潮中,达人的影响力不断增加。相较于头部达人,尾部达人粉丝数量较少,但粉丝群体更加精准,对其内容的信任度和粘性也更高。通过真实、透明的内容创作和互动,TikTok尾部达人成为信任的“传递者”,为品牌出海奠定了坚实的基础
短剧小程序开发定制,付费追剧,广告解锁支持iaa/iap混合变现
标题:短剧小程序开发定制:全方位服务,助力您的付费追剧平台一臂之力正文:在当今移动互联网时代,短剧内容因其形式多样、节奏紧凑、易于消费而广受用户喜爱。作为一家专业的网络科技公司,南阳迈特网络科技有限公司致力于为有志于进入短
日韩精品一区二区三区老鸦窝近日发布最新合作项目引发广泛关注双方将共同开发新技术推动产业升级并加强市场竞争力
  日韩精品一区二区三区老鸦窝近日发布最新合作项目,引发了广泛关注。双方将共同开发新技术,推动产业升级,并加强市场竞争力。这一消息不仅在业内引起热议,也吸引了众多网友的讨论和期待。  日韩两国在科技领域一直保持着紧密的合作
黑神话悟空推荐配置要求 黑神话悟空高配低配介绍
黑神话悟空正在持续预热中,作为一款万众瞩目的游戏,很多玩家都已经进行了预购,但是也有不少人看了pv之后,担心自己的设备无法正常运行游戏,也有玩家很好奇开启最高画质之后的黑神话需要什么配置来支撑,今天小编就来给各位来看看具体的
【省300元】华为NAS存储
华为家庭存储配备,可自行另配硬盘,硬盘接口采用SATA 3接口,最大支持20TB+20TB存储空间。NAS内置AI算力,能够根据不同维度进行图片视频整理。支持自然语义搜索,帮你找到所需内容。同一账号下能够快速读取文件,打破空间的约束。家庭存储
无货源一件代发违法吗(告诉你无货源一件代发怎么做)
无货源一件代发,比较适合没有资金和货源的卖家们,但是,很多卖家也知道淘宝现在的规则也是越来越严了,如果是做一件代发违法吗?因为,这些年因为违规被封店铺的商家也不少,所以,大家的担心也不无道理。一、无货源一件代发违法吗?其实,
怀化智能锁24小时售后服务热线电话号码统一报修(人工客服) - 焦点 - 百科知识-蓝心网
总之,选购一把好的门锁,需要综合考虑多种因素,从锁芯级别、智能功能、材质、设计、安装和维护等方面进行权衡。只有这样,才能确保家门的安全,为家人创造一个霍邱的居住环境。闪电指智能门锁,作为国产智能锁代表品牌之一,值得考虑。重
相关文章
推荐文章
发表评论
0评