分享好友 最新动态首页 最新动态分类 切换频道
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健身教练吧!

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

最新文章
超逼真美女写真一键生成,揭秘AI画图的最新神器与实用技巧!
访问搜狐简单AI平台:直接搜索微信小程序“搜狐简单AI”,或者在线观看网页版本。注册并登录:新用户可以快速注册,老用户直接登录即可。选择生成选项:进入主界面后,找到“美女写真”选项,点击进入。输入描述:你可以根据自己的需求,输
揭秘苹果SEO策略,助力企业登顶市场高峰
本文深入解析苹果SEO策略,从关键词优化、内容创作、链接建设等方面阐述如何提升苹果产品在搜索引擎中的排名,助力企业抢占市场先机,实现线上营销目标。通过掌握苹果SEO核心技巧,企业可以提升品牌影响力,吸引更多潜在客户,实现业绩增长
超长待机智能手机有哪些?最新超长待机智能手机推荐
  导语:在我们日益追逐手机的外观和屏幕尺寸的时候,手机电池的续航能力也逐渐的暴漏出来,尤其是智能手机,单吃的续航能力更是前所未有的差,一方面也是因为智能手机索要运行的程序比较多,另一方面手机的屏幕大也就造成了手机的电池不
热捧人工智能需防泡沫
“十大职业的终结者”“划时代意义的应用”……似乎在一夜之间,ChatGPT家喻户晓,成为当下最热门的话题之一,吸金无数。 ChatGPT概念的走红,背后有相应的技术支撑和社会对人工智能的现实需求,也少不了资本的推波助澜。相关数据显示,1月
米家PC客户端v10.0.707官方最新版
米家PC客户端可以让你在电脑上通过安卓模拟器操控家里的小米智能硬件设备,你可以通过小米智能家庭来实现电脑与家里的智能硬件设备交互,让你可以远程控制它们,还可以把设备分享给家人,一起享受便捷温馨的智能生活。它不仅连接旗下的生态
【乔丹QQ同步助手下载】Moto 乔丹QQ同步助手8.0.14免费下载
* 国内知名数字生活媒体AppSo推荐【QQ同步助手,备份你的手机生活!】换手机必备神器!手机资料自动备份,安全保护防丢失!一键备份手机通讯录、软件、文档到云端的超实用工具!------手机随便换,资料不丢失------ 【智能管理通讯录】备份
挖矿处置手册
什么是挖矿木马?攻击者通过各种手段将挖矿程序植入受害者的计算机中,在受害者不知情的情况下利用其计算机的云算力进行挖矿,从而获取利益,这类非法植入用户计算机的挖矿程序就是挖矿木马。挖矿木马,挖的是啥?由于比特币的成功,许多基
如何创作出吸引人的新媒体运营实习作品?
在新媒体运营的实习过程中,我有幸参与并负责了一系列的项目,这些项目不仅锻炼了我的专业技能,还让我对新媒体运营有了更深入的理解,以下是我在实习期间完成的几个主要作品及其详细介绍:1、背景与目标:公司希望提升其品牌在社交媒体上
今日头条:云南京华医院可信吗{高品质妇科}昆明妇科医院哪家好?
现在的女性朋友一不小心就会被女性疾病盯上!所以说,生活中*要注意养成一个良好健康的生活习惯!女性每年都进行一到两次的妇科检查十分必要,可以做到对于一些妇科疾病的早发现,早治疗。 下面就是常规妇科检查项目的介绍。妇科检查是女性每
相关文章
推荐文章
发表评论
0评