爱笑的小姐姐 · 2021年11月17日

OpenGait:首个步态识别框架开源了!

远远地,看一眼你走路方式,就知道你是谁!这就是步态识别技术。
image.png
首个步态识别框架OpenGait正式发布了!OpenGait由南方科技大学计算机系于仕琪科研团队开发,银河水滴公司予以支持。欢迎大家三连:使用,反馈和建议。项目主页:https://github.com/ShiqiYu/Op...

在复现步态算法或者设计识别算法时,相信大家都遇到过相同的问题:当你准备复现顶会论文,却需要从头搭建实验环境,手撸训练,评估代码。勤勤恳恳的写完后,又往往bug频出。当一些优秀的顶会算法开源后,复用代码时又修改得晕头转向。

OpenGait通过模块化的设计,代码高效且可复用,助你轻松复现论文,调参炼丹的路上,OpenGait值得你拥有!

第一版的开源步态识别框架OpenGait,主要特点如下:

  • 我们提供了一个高效简单的BaseLine模型,搭配了tricks后,性能媲美SOTA
  • OpenGait复现了4篇顶会的SOTA算法,取得相近甚至更高的结果
  • 支持分布式多卡训练,在大数据吞吐量的情况下提速20%以上;分布式多卡测试,跑完CASIA-B仅需15秒!(5485个序列,BaseLine,2 1080ti gpus),跑完OUMVLP仅需不到两分钟!(133780个序列,Baseline,8 1080ti gpus)。注:OUMVLP尚未整理,预计在下一版本推出,在当前版本也可轻松拓展,欢迎有能力的同学多多贡献代码!
  • 支持混合精度训练,当gpu支持半精度时显存直接砍半,并提速20%~50%(视具体GPU型号而定)。
  • 友好的可视化训练过程,支持Tensorboard可视化及日志记录,可自定义添加可视化内容。
  • 支持多阶段训练及边训边测
  • 模块化设计,常规模型可以仅通过配置文件定义,主流的操作也可以通过配置文件客制化加载。
  • 诸如优化器,调度器,损失函数,采样器等均可通过配置文件一键式部署。
  • 支持组合损失函数。

支持的模型如下(括号里为原论文准确率):

image.png
当前版本仅支持剪影(silhouette)数据集的训练测试。我们期望能收到各位反馈,在公众号评论区,邮件,或者github中提出issue,留下你们迫切期望集成的功能,我们在下一个版本中更新。

OpenGait不仅仅是为了提供一个高效可拓展的步态研究框架,更多的是提供一个相对公平的评估框架,帮助模型发挥其真正的潜力。

主要贡献人:

  • 博士生樊超
  • 博士生沈川福
  • 硕士生梁峻豪

特别感谢:

  • 专注于步态识别技术的高科技公司银河水滴科技(北京)有限公司
原文链接:https://mp.weixin.qq.com/s/5WTJ2WD_DJdyPl-D1l6vvw
作者:于仕琪团队

推荐阅读

推荐阅读
关注数
18790
内容数
1342
嵌入式端AI,包括AI算法在推理框架Tengine,MNN,NCNN,PaddlePaddle及相关芯片上的实现。欢迎加入微信交流群,微信号:aijishu20(备注:嵌入式)
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息