标题:一种针对视频的人体姿态估计的加速算法 作者:王俊杭 陈贝佳 邵家玉 作者单位:东南大学自动化学院,江苏 南京 210096 关键字:人体姿态估计;高分辨率网络;推理速度;网络结构 摘要:为了解决视频的高性能人体姿态估计算法参数量和计算量庞大导致的推理速度慢的问题,提出了基于高分辨率网络(HRNet)的人体姿态估计改进算法。该算法在检测过程中采用隔帧检测和去抖动的优化处理,优化人体检测流程;针对姿态估计网络,使用ShuffleUnitV2组件对 HRNet 重新设计得到了S-HRNet,提高网络的利用率。实验结果表明:在公开数据集COCO训练集上,改进算法的总推理时间为356 ms,而原始算法总推理时间为992 ms,有效地提高推理速度。改进后的算法解决了原有的 HRNet 模型参数量大、推理速度慢的问题,同时也保持了一定的性能,为实际部署提供了一个适合的算法。 |